transformer_engine/paddle/csrc/custom_ops.cu · 08b4997605e99c911664539b33657d0c03fe27a6 · OpenDAS / TransformerEngine

[Paddle] Fix device memory leak (#1029) · 08b49976

Tian Zheng authored Jul 25, 2024



* i
Signed-off-by: Tian Zheng (Engrg-Hardware 1) <tizheng@nvidia.com>

* .
Signed-off-by: Tian Zheng (Engrg-Hardware 1) <tizheng@nvidia.com>

---------
Signed-off-by: Tian Zheng (Engrg-Hardware 1) <tizheng@nvidia.com>

08b49976

custom_ops.cu 86.4 KB

Replace custom_ops.cu