commit f99adbb4ec7a738684925d111f81f6b523cc83d0
author Peter Bell <peterbell10@live.co.uk> Mon Jan 29 18:43:02 2024 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Mon Jan 29 20:23:40 2024 +0000
tree 8d789fd9087c05be1898157359ea8b9958eb905e
parent 65917411834874a98d39923c179a1308d371a733

[inductor] Remove ROCm xfail on test_cum{sum,prod}_zero_dim (#118558)

Fixes #118540, fixes #118541

Since the zero-dim case reduces to a pointwise operation, we don't fallback on
ROCm.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/118558
Approved by: https://github.com/malfet

test/inductor/test_torchinductor_codegen_dynamic_shapes.py

diff --git a/test/inductor/test_torchinductor_codegen_dynamic_shapes.py b/test/inductor/test_torchinductor_codegen_dynamic_shapes.py
index af25284..33761b8 100644
--- a/test/inductor/test_torchinductor_codegen_dynamic_shapes.py
+++ b/test/inductor/test_torchinductor_codegen_dynamic_shapes.py
@@ -293,9 +293,7 @@
 if TEST_WITH_ROCM:
     test_failures.update(
         {
-            "test_cumprod_zero_dim_dynamic_shapes": TestFailure(("cpu", "cuda")),
             "test_cumsum_dynamic_shapes": TestFailure(("cpu", "cuda")),
-            "test_cumsum_zero_dim_dynamic_shapes": TestFailure(("cpu", "cuda")),
         }
     )