diff --git a/test/spmd/test_spmd_lowering_context.py b/test/spmd/test_spmd_lowering_context.py index 2589e060ab9..46fb6a23a80 100644 --- a/test/spmd/test_spmd_lowering_context.py +++ b/test/spmd/test_spmd_lowering_context.py @@ -100,8 +100,6 @@ def fn(x, y): assert expected_output[1] == f"f32[32,2048] {b_sharding_spec}" assert expected_output[2] == f"f32[2048] {a_sharding_spec}" assert expected_output[3] == f"f32[32,2048] {b_sharding_spec}" - self.assertTrue(met.counter_value("ExecuteReplicated") > 0) - self.assertTrue(met.counter_value("ExecuteComputation") is None) def test_device_parameter_id_tensor_mapping(self): met.clear_all()