non-deterministic IMA on B100 `tests.python.direct.test_cutlass_gemm.test_grouped_mm`

[liqiangxl]

reported by @jjsjann123
jit_python_tests_20_B100
I'm seeing non-deterministic IMA popping up pretty frequently in dashboard for tests.python.direct.test_cutlass_gemm.test_grouped_mm