Merge branch 'custom_op_infer_smpd' of https://github.com/liuzhenhai9…

…3/Paddle into custom_op_infer_smpd
PaddlePaddle · Jan 6, 2024 · 548dd89 · 548dd89
2 parents 32368a7 + dd4c8d4
commit 548dd89
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/test/auto_parallel/custom_op/test_semi_auto_parallel_custom_op.py b/test/auto_parallel/custom_op/test_semi_auto_parallel_custom_op.py
@@ -18,7 +18,6 @@
 
 import collective.test_communication_api_base as test_base
 
-from paddle.framework import core
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 
 
@@ -48,12 +47,6 @@ def test_dynamic_auto_parallel(self):
                 user_defined_envs=envs,
             )
 
-    # test static spmd rule register
-    def test_static_rule(self):
-        import custom_relu  # noqa: F401 # pylint: disable=unused-import
-
-        assert core.contains_spmd_rule("custom_relu")
-
 
 if __name__ == "__main__":
     unittest.main()