[BE] remove old pytorch version warning on strided sharding since 2.5…

… is official released [ghstack-poisoned]
pytorch · Oct 30, 2024 · ab922b6 · ab922b6
1 parent 53d0f69
commit ab922b6
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 36 deletions.
diff --git a/torchtitan/parallelisms/parallelize_llama.py b/torchtitan/parallelisms/parallelize_llama.py
@@ -330,12 +330,6 @@ def apply_fsdp(
     if cpu_offload:
         fsdp_config["offload_policy"] = CPUOffloadPolicy()
 
-    # TODO: remove this check once PyTorch 2.5 is released. We can safely assume
-    # that users won't use a nightly build which is older than 20240809 by then.
-    if tp_enabled:
-        # check if strided sharding is enabled, which is necessary for 2D/3D DCP
-        check_strided_sharding_enabled()
-
     for layer_id, transformer_block in model.layers.items():
         if pp_enabled:
             # For PP, do not reshard after forward to avoid per-microbatch

diff --git a/torchtitan/parallelisms/utils.py b/torchtitan/parallelisms/utils.py