From d16e2d834f03ddd5bde217d6cf918e5e767a55df Mon Sep 17 00:00:00 2001 From: Andrey Bokovoy Date: Wed, 15 Jan 2025 18:42:54 -0800 Subject: [PATCH] Enable v2 forward test for ROCm (#3573) Summary: X-link: https://github.com/facebookresearch/FBGEMM/pull/663 Since https://github.com/pytorch/FBGEMM/pull/3266 is merged, the v2 forward kernel should be tested for ROCm devices as well. cc: liligwu Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/3573 Reviewed By: leitian, spcyppt Differential Revision: D68237138 Pulled By: q10 fbshipit-source-id: 3f743303c13ae79976dda273ffb75deee57fd11f --- fbgemm_gpu/test/tbe/training/forward_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fbgemm_gpu/test/tbe/training/forward_test.py b/fbgemm_gpu/test/tbe/training/forward_test.py index bb519864d6..1832db0ef8 100644 --- a/fbgemm_gpu/test/tbe/training/forward_test.py +++ b/fbgemm_gpu/test/tbe/training/forward_test.py @@ -541,7 +541,7 @@ def test_forward_gpu_no_cache_fp16( @unittest.skipIf(*gpu_unavailable) @given( - use_experimental_tbe=st.booleans() if not TEST_WITH_ROCM else st.just(False), + use_experimental_tbe=st.booleans(), ) @settings( verbosity=VERBOSITY, @@ -670,7 +670,7 @@ def test_forward_gpu_uvm_cache_int8( @unittest.skipIf(*gpu_unavailable) @given( cache_algorithm=st.sampled_from(CacheAlgorithm), - use_experimental_tbe=st.booleans() if not TEST_WITH_ROCM else st.just(False), + use_experimental_tbe=st.booleans(), ) @settings( verbosity=VERBOSITY, @@ -740,7 +740,7 @@ def test_forward_gpu_uvm_cache_fp16( @unittest.skipIf(*gpu_unavailable) @given( cache_algorithm=st.sampled_from(CacheAlgorithm), - use_experimental_tbe=st.booleans() if not TEST_WITH_ROCM else st.just(False), + use_experimental_tbe=st.booleans(), ) @settings( verbosity=VERBOSITY,