From a414a3bde005afb195f3fbfaca1c5f15bd313557 Mon Sep 17 00:00:00 2001
From: Chien-Chin Huang <chienchin@fb.com>
Date: Tue, 10 Sep 2024 10:48:38 -0700
Subject: [PATCH] Update

[ghstack-poisoned]
---
 test/test_fused_rms_norm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_fused_rms_norm.py b/test/test_fused_rms_norm.py
index 9bd7e3732c..d5c353c2f1 100644
--- a/test/test_fused_rms_norm.py
+++ b/test/test_fused_rms_norm.py
@@ -11,7 +11,7 @@
     Replicate,
     Shard,
 )
-from torch.distributed._tensor.debug import CommDebugMode
+from torch.distributed.tensor.debug import CommDebugMode
 from torch.testing._internal.common_utils import run_tests
 from torch.testing._internal.distributed._tensor.common_dtensor import (
     DTensorTestBase,