[ROCm] Use tl.range() in block GEMM kernels with num_stages set by host.

[ROCm] Use tl.range() in block GEMM kernels with num_stages set by host. #9052