transformer_engine/pytorch/graph.py · 697b52cbde6a2d2b67f71879b63203d5082e1f8c · OpenDAS / TransformerEngine

Robin Zhang authored Dec 12, 2025



Convert sample tuple to list in reuse
Signed-off-by: Robin Zhang <robinz@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

50352325

graph.py 57.3 KB