fix(tests): run `causal_softmax` reference on CPU#612

Merged

voltjia merged 1 commit into

fix/causal-softmax-reference

May 17, 2026

Commits on May 16, 2026

fix(tests): run causal softmax reference on CPU
voltjia
committed