Fix AMD test running on nvidia acidentally (#4844)

Summary:
Pull Request resolved: #4844

X-link: facebookresearch/FBGEMM#1873

as titled, missed this originally

Reviewed By: q10

Differential Revision: D82043068

fbshipit-source-id: 8440759c3c3c8ad5f5a7bbe75ee501180ce79e95

Author: cthi

fbgemm_gpu/experimental/gen_ai/test/quantize/quantize_test.py

@@ -1166,7 +1166,9 @@ def bf16_loopover_validate(
         N=st.sampled_from([256, 1024, 6144]),
         K=st.sampled_from([256, 512, 3584]),
         use_cudagraph=st.booleans(),
-        mode=st.sampled_from(["stacked", "torch_2d3d"]),
+        mode=st.sampled_from(
+            ["stacked"] + (["torch_2d3d"] if torch.version.hip else [])
+        ),
     )
     def test_grouped_gemm_2d_3d(
         self,