transformer_engine/pytorch/ops/sequential.py · 9c127ef5b1ee10d466acf633f4a0ad3c8914cf1b · OpenDAS / TransformerEngine

[PyTorch] Update Sequential container to handle changes in module base class (#1028) · 33a3d02f

Tim Moon authored Jul 20, 2024



* Update sequential container constructor to handle modules in plain dicts
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Avoid initializing Sequential with dicts
Signed-off-by: Tim Moon <tmoon@nvidia.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>

33a3d02f

sequential.py 5.63 KB

Replace sequential.py