[v1] support training with fsdp2 (#9773)

Co-authored-by: frozenleaves <frozen@Mac.local>
Co-authored-by: Yaowei Zheng <hiyouga@buaa.edu.cn>

tests_v1/plugins/model_plugins/test_kernel_plugin.py

@@ -15,7 +15,6 @@
 import sys
 from unittest.mock import MagicMock, patch
 
-import pytest
 import torch.multiprocessing as mp
 from transformers import AutoModelForCausalLM