[BUG] RuntimeError: CUDA error: operation not supported

**Describe the bug**
```
[rank0]: Traceback (most recent call last):
[rank0]: File "/root/miniconda3/envs/swift/lib/python3.12/site-packages/swift/cli/export.py", line 5, in
[rank0]: export_main()
[rank0]: File "/root/miniconda3/envs/swift/lib/python3.12/site-packages/swift/llm/export/export.py", line 53, in export_main
[rank0]: return SwiftExport(args).main()
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/root/miniconda3/envs/swift/lib/python3.12/site-packages/swift/llm/base.py", line 49, in main
[rank0]: result = self.run()
[rank0]: ^^^^^^^^^^
[rank0]: File "/root/miniconda3/envs/swift/lib/python3.12/site-packages/swift/llm/export/export.py", line 37, in run
[rank0]: convert_hf2mcore(args)
[rank0]: File "/root/miniconda3/envs/swift/lib/python3.12/site-packages/swift/megatron/utils/convert.py", line 178, in convert_hf2mcore
[rank0]: initialize_megatron(extra_args_provider=extra_args_provider, args_defaults=extra_args)
[rank0]: File "/root/paddlejob/workspace/env_run/Megatron-LM/megatron/training/initialize.py", line 156, in initialize_megatron
[rank0]: finish_mpu_init()
[rank0]: File "/root/paddlejob/workspace/env_run/Megatron-LM/megatron/training/initialize.py", line 138, in finish_mpu_init
[rank0]: MoEAuxLossAutoScaler.set_loss_scale(torch.ones(1, device=torch.cuda.current_device()))
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: RuntimeError: CUDA error: operation not supported
[rank0]: CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
[rank0]: For debugging consider passing CUDA_LAUNCH_BLOCKING=1
[rank0]: Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
```

**To Reproduce**
https://swift.readthedocs.io/en/latest/Instruction/Megatron-SWIFT-Training.html
```
CUDA_VISIBLE_DEVICES=0 \
swift export \
--model Qwen/Qwen2.5-7B-Instruct \
--to_mcore true \
--torch_dtype bfloat16 \
--output_dir Qwen2.5-7B-Instruct-mcore \
--test_convert_precision true
```

**Environment (please complete the following information):**
- Megatron-LM commit ID Megatron-LM.git@core_r0.13.0
```
>>> import torch

>>>
>>> print(torch.cuda.nccl.version())
(2, 26, 2)
>>> print(torch.__config__.show())
PyTorch built with:
- GCC 11.2
- C++ Version: 201703
- Intel(R) oneAPI Math Kernel Library Version 2024.2-Product Build 20240605 for Intel(R) 64 architecture applications
- Intel(R) MKL-DNN v3.7.1 (Git Hash 8d263e693366ef8db40acc569cc7d8edf644556d)
- OpenMP 201511 (a.k.a. OpenMP 4.5)
- LAPACK is enabled (usually provided by MKL)
- NNPACK is enabled
- CPU capability usage: AVX512
- CUDA Runtime 12.6
- NVCC architecture flags: -gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75;-gencode;arch=compute_80,code=sm_80;-gencode;arch=compute_86,code=sm_86;-gencode;arch=compute_90,code=sm_90
- CuDNN 90.5.1
- Magma 2.6.1
- Build settings: BLAS_INFO=mkl, BUILD_TYPE=Release, COMMIT_SHA=e2d141dbde55c2a4370fac5165b0561b6af4798b, CUDA_VERSION=12.6, CUDNN_VERSION=9.5.1, CXX_COMPILER=/opt/rh/gcc-toolset-11/root/usr/bin/c++, CXX_FLAGS= -D_GLIBCXX_USE_CXX11_ABI=1 -fvisibility-inlines-hidden -DUSE_PTHREADPOOL -DNDEBUG -DUSE_KINETO -DLIBKINETO_NOROCTRACER -DLIBKINETO_NOXPUPTI=ON -DUSE_FBGEMM -DUSE_PYTORCH_QNNPACK -DUSE_XNNPACK -DSYMBOLICATE_MOBILE_DEBUG_HANDLE -O2 -fPIC -Wall -Wextra -Werror=return-type -Werror=non-virtual-dtor -Werror=range-loop-construct -Werror=bool-operation -Wnarrowing -Wno-missing-field-initializers -Wno-unknown-pragmas -Wno-unused-parameter -Wno-strict-overflow -Wno-strict-aliasing -Wno-stringop-overflow -Wsuggest-override -Wno-psabi -Wno-error=old-style-cast -fdiagnostics-color=always -faligned-new -Wno-maybe-uninitialized -fno-math-errno -fno-trapping-math -Werror=format -Wno-stringop-overflow, LAPACK_INFO=mkl, PERF_WITH_AVX=1, PERF_WITH_AVX2=1, TORCH_VERSION=2.7.1, USE_CUDA=ON, USE_CUDNN=ON, USE_CUSPARSELT=1, USE_GFLAGS=OFF, USE_GLOG=OFF, USE_GLOO=ON, USE_MKL=ON, USE_MKLDNN=ON, USE_MPI=OFF, USE_NCCL=1, USE_NNPACK=ON, USE_OPENMP=ON, USE_ROCM=OFF, USE_ROCM_KERNEL_ASSERT=OFF,
```

Start with the provided export command and traceback, then inspect megatron/training/initialize.py around finish_mpu_init and swift/megatron/utils/convert.py around convert_hf2mcore. Reproduce the CUDA failure with the listed PyTorch, CUDA, and Megatron-LM versions, and confirm that the export with --to_mcore and --test_convert_precision completes successfully.

Written by the indexing model from the issue text.