`use_cpu=st.booleans()` in `jagged_index_select_2d_ref` fails the test on ROCm

[Li Li]

The test jagged_index_select_2d_ref fails if use_cpu=True on ROCm,

FBGEMM/fbgemm_gpu/test/jagged_tensor_ops_test.py

Line 1469 in 8616ed7

         use_cpu=st.booleans() if gpu_available else st.just(True), 

The issue is akin to

FBGEMM/fbgemm_gpu/test/split_table_batched_embeddings_test.py

Line 1009 in 8616ed7

         if (gpu_available and not TEST_WITH_ROCM) 

A PR is opened to disable the CPU test on ROCm
#1635

[Benson Ma]

@liligwu Is the issue considered resolved now that #1635 has merged?

[Li Li]

@liligwu Is the issue considered resolved now that #1635 has merged?

Yes, I'm closing it. Thank you for reminding me.