3a5cc2a589de0b7cc601a70f9b11cd3884466abb
- _custom_ops.py: 改用importlib.util加载.so替代torch.ops.load_library - patch_ops.sh: docker build时编译moe_topk_softmax_v3 CUDA kernel - 真机验证: weights sum=1.0, GPU test passed
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%