transformer_engine/pytorch/module/base.py · 324be3324278723bd8f66196ed1ccac29b94bd7f · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

Reset cache logic of weight workspace for NVFP4TensorStorage (#2524) · dbd0197e
Jinhang Choi authored Dec 16, 2025
```
reset weight ws cache for NVFP4TensorStorage
Signed-off-by: Jinhang Choi <jinhangc@nvidia.com>
```
dbd0197e

base.py 70 KB