transformer_engine/pytorch/graph.py · 161b1d98f80243c78ddecdadd15b010549e4e3d0 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[PyTorch] Add pool argument to make_graphed_callable (#1218) · 728c558b
Kirthi Shankar Sivamani authored Sep 30, 2024
```
Add pool argument to make_graphed_callable
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
728c558b

graph.py 27.6 KB