Update on "[6/N][BE] Remove Sharded Linear Op for ShardedTensor"

fduwjj · fduwjj · commit a9d6b23aa520 · 2023-03-06T02:22:03.000Z
[ghstack-poisoned]
diff --git a/torch/distributed/_shard/sharded_tensor/_ops/__init__.py b/torch/distributed/_shard/sharded_tensor/_ops/__init__.py
@@ -9,7 +9,6 @@
 from .init import kaiming_uniform_, normal_, uniform_, constant_
 
 # Import all ChunkShardingSpec ops
-from torch.distributed._shard.sharding_spec.chunk_sharding_spec_ops.linear import sharded_linear
 from torch.distributed._shard.sharding_spec.chunk_sharding_spec_ops.embedding import sharded_embedding
 from torch.distributed._shard.sharding_spec.chunk_sharding_spec_ops.embedding_bag import sharded_embedding_bag
 from torch.distributed._shard.sharding_spec.chunk_sharding_spec_ops.softmax import sharded_softmax
