transformer_engine/paddle/cpp_extensions.py · d793ca17ac3b1206f420f273c1747e648cde96f6 · OpenDAS / TransformerEngine

Shijie authored Jul 31, 2024



add deterministic option
Signed-off-by: Shijie Wang <jaywan@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

d793ca17

cpp_extensions.py 33.1 KB