project6
8b6f3fd242
fix(MoE): robust CUDA kernel loading + no-GPU precompile
1. precompile_moe_topk.py: skip GPU verification during Docker build
(torch.cuda.is_available() check — .so compilation doesn't need GPU)
2. _custom_ops.py topk_softmax init: 3-tier loading
- import precompiled module (torch cache)
- scan known .so paths (torch_extensions cache dirs)
- JIT compile from .cu source
- PyTorch fallback with WARNING (not silent — must know if CUDA failed)
3. patch_ops.sh: report .so location after precompile for debugging
2026-08-10 07:50:28 +00:00
..
2026-08-10 06:34:54 +00:00
2026-08-10 03:59:45 +00:00
2026-08-10 06:36:40 +00:00
2026-08-10 02:55:58 +00:00
2026-08-10 04:01:21 +00:00
2026-08-10 07:50:28 +00:00