transformer_engine/common/util/cast_kernels.cuh · 2a95efd39128955081c60b67d49351d89f003324 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[common] Removed tensor boundary checks in MXFP8 kernels (#1519) · bc4c452c
Oleg Goncharov authored Mar 04, 2025
```
Added constexpr checks of tensor boundaries
Signed-off-by: Oleg Goncharov <ogoncharov@nvidia.com>
```
bc4c452c

cast_kernels.cuh 53.8 KB