Xingzhi107

When I use a custom expert, inherit the FMoE class, and turn on Smart schedule to report an error ``` [ubuntu:8399 :0:8399] Caught signal 11 (Segmentation fault: invalid permissions for...

### 🐛 Describe the bug When I run examples/language/gpt/experiments/auto_parallel/auto_parallel_with_gpt.py,The error is as follows File "/opt/conda/lib/python3.9/site-packages/colossalai/auto_parallel/tensor_shard/initialize.py", line 265, in initialize_model gm = ColoGraphModule(model, graph, model.__class__.__name__) File "/opt/conda/lib/python3.9/site-packages/colossalai/_analyzer/fx/graph_module.py", line 110, in __init__...