Files
tilelang/testing/python/cuda
Chennes c9e5112332 [BugFix] Fix bf16 CUDA exp self-recursion (#2402)
`hexp` is #define'd to `cutlass::fast_exp`, so the hand-written
`cutlass::fast_exp(bfloat16_t)` overload's `return ::hexp(x)` expanded into a
call to itself: bf16 `T.exp` recursed and returned garbage.

Route through float, mirroring cutlass's own `half_t fast_exp`
(cutlass/fast_math.h), so the call resolves to `cutlass::fast_exp(float)`
(= ::expf). Verified against torch.exp using the #2383 repro.

Fixes #2383.
2026-06-15 21:21:32 +08:00
..