rename sym mem buffer file

Author: danielvegamyhre

benchmarks/prototype/moe_training/mxfp8/ep/syncless/bench_moe_e2e.py

@@ -47,7 +47,7 @@
     MXFP8TrainingOpConfig,
     MXFP8TrainingRecipe,
 )
-from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
     SymmetricMemoryBufferManager,
 )
 from torchao.prototype.moe_training.ep.syncless.expert_parallel import (

benchmarks/prototype/moe_training/mxfp8/ep/syncless/bench_token_combine.py

@@ -28,7 +28,7 @@
 
 from benchmarks.utils import profile_fn
 from torchao.prototype.moe_training.ep.permute import permute_and_pad
-from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
     SymmetricMemoryBufferManager,
 )
 from torchao.prototype.moe_training.ep.syncless.token_combine import token_combine

test/prototype/moe_training/ep/syncless/test_dispatch_with_managed_buffer.py

@@ -148,7 +148,7 @@ def test_save_activations_in_managed_buffer(self):
                 total_global_tokens + 128 * experts_per_rank * self.world_size
             )
 
-            from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+            from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
                 get_buffer_manager,
             )
 

test/prototype/moe_training/ep/syncless/test_token_combine.py

@@ -94,7 +94,7 @@ def test_token_combine(self):
                 total_global_tokens + 128 * experts_per_rank * self.world_size
             )
 
-            from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+            from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
                 get_buffer_manager,
             )
 

test/prototype/moe_training/ep/syncless/test_token_dispatch.py

@@ -105,7 +105,7 @@ def test_mxfp8_a2a(self):
             )
 
             # Preallocate symmetric memory buffers
-            from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+            from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
                 get_buffer_manager,
             )
 

torchao/prototype/moe_training/ep/syncless/__init__.py

@@ -1,4 +1,4 @@
-from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
     SymmetricMemoryBufferManager,  # noqa: F401
     get_buffer_manager,  # noqa: F401
 )

torchao/prototype/moe_training/ep/syncless/expert_parallel.py

@@ -32,7 +32,7 @@
 from torch.distributed.tensor import Shard
 from torch.distributed.tensor.parallel import ParallelStyle
 
-from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
     SymmetricMemoryBufferManager,
     get_buffer_manager,
 )

…e_training/ep/syncless/buffer_manager.py …ng/ep/syncless/sym_mem_buffer_manager.pytorchao/prototype/moe_training/ep/syncless/buffer_manager.py renamed to torchao/prototype/moe_training/ep/syncless/sym_mem_buffer_manager.py torchao/prototype/moe_training/ep/syncless/buffer_manager.py renamed to torchao/prototype/moe_training/ep/syncless/sym_mem_buffer_manager.py

@@ -23,7 +23,7 @@
 )
 from torch.library import triton_op, wrap_triton
 
-from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
     SymmetricMemoryBufferManager,
 )
 
@@ -65,7 +65,7 @@ def forward(
             buffer_manager: optional buffer manager for reusing buffers.
             token_alignment: expert token group alignment (default 128).
         """
-        from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+        from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
             get_buffer_manager,
         )
 

torchao/prototype/moe_training/ep/syncless/token_combine.py

@@ -4,7 +4,7 @@
 import triton.language as tl
 from torch.library import triton_op, wrap_triton
 
-from torchao.prototype.moe_training.ep.syncless.buffer_manager import (
+from torchao.prototype.moe_training.ep.syncless.sym_mem_buffer_manager import (
     SymmetricMemoryBufferManager,
     get_buffer_manager,
 )