transformer_engine/common/transpose/transpose.cu · d99142a0177a2462cbda07a31aaa8e68b4e85461 · OpenDAS / TransformerEngine

"cmd/git@developer.sourcefind.cn:liming6/sshd-tool.git" did not exist on "abfc4893f9d2abe9908fb0e407e5f67de1d0fce6"

NVRTC kernels for cast-transpose (#258) · 14c1ecd0

Tim Moon authored Apr 19, 2024



* Add NVRTC kernels for cast-transpose
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Update copyright year
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Add noop flag to NVRTC cast-transpose kernel
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Apply suggestions from code review
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

14c1ecd0

transpose.cu 12.6 KB

Replace transpose.cu