Commit 04f0cd6

committed

Fix: Resolve parameter mismatch between TE_FL and NVTE functions

1 parent b0a5934 commit 04f0cd6Copy full SHA for 04f0cd6

6 files changed

transformer_engine
- plugin/core
  - backends/vendor/cuda
    - cuda.py
  - ops.py
- pytorch
  - module
    - layernorm_linear.py
  - ops/basic
    - rmsnorm.py
  - optimizers
    - __init__.py
    - fused_adam.py

Comments

(0)