From 29c443b7f10ab57c117bc97eb90211a06f1fc3bc Mon Sep 17 00:00:00 2001 From: Wonjoo Lee Date: Wed, 13 Mar 2024 04:05:57 +0000 Subject: [PATCH] Update existing unit tests for aten::add HLO dumps --- test/spmd/test_xla_sharding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/spmd/test_xla_sharding.py b/test/spmd/test_xla_sharding.py index 06dbab66fc7..46e12e383be 100644 --- a/test/spmd/test_xla_sharding.py +++ b/test/spmd/test_xla_sharding.py @@ -831,7 +831,7 @@ def test_mark_sharding_ir(self): actual += 0 hlo = torch_xla._XLAC._get_xla_tensors_hlo([actual.global_tensor]) self.assertIn( - '%add.12 = f32[1,128]{1,0} add(f32[1,128]{1,0} %custom-call.10, f32[1,128]{1,0} %broadcast.11)', + '%add.12 = f32[1,128]{1,0} add(f32[1,128]{1,0} %custom-call.9, f32[1,128]{1,0} %broadcast.11)', hlo) self.assertTrue(torch.allclose(expected, actual.cpu()))