transformer_engine/common/transpose/cast_transpose.cu · 166bb0787da431b4bde2d7aa65aee4022a9976ed · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

Reduce CUDA driver calls when choosing transpose kernels (#1002) · 166bb078
Tim Moon authored Jul 10, 2024
```
Reduce CUDA driver API calls when choosing transpose kernels
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
166bb078

cast_transpose.cu 15.7 KB