add fsdp2 e2e tests

This commit is contained in:
Wing Lian
2025-04-01 19:00:21 -04:00
parent 252dc5c91b
commit 2b5e546da0
2 changed files with 71 additions and 0 deletions

View File

@@ -42,6 +42,7 @@ class LigerPlugin(BasePlugin):
def pre_model_load(self, cfg):
if cfg.torch_compile:
# torch compile will unnecessarily attempt to optimize the triton kernel unless explicitly disabled
import liger_kernel.ops.fused_linear_cross_entropy
patch_with_compile_disable(