project6-dev 7969874cde fix: cast router_logits to float32 before topk kernel
router_logits from ReplicatedLinear is float16. PyTorch path does
.float() but kernel path passed float16 directly. BI-V100 corex CUB
half-precision softmax produces garbage. Cast to float32 first.
2026-08-12 10:06:19 +00:00
2026-08-07 08:43:51 +00:00
2026-07-30 17:03:23 +08:00
2026-08-11 01:22:31 +00:00

project_6

Description
No description provided
Readme 428 MiB
Languages
C++ 41.8%
Cuda 31.6%
Python 22.2%
C 2.1%
CMake 1.1%
Other 1.1%