Commits · 782101277c90fa3ae800bb31b6465a318f0d954c · OpenDAS / TransformerEngine

03 Nov, 2022 2 commits

Conditional wgrad support (#21) · 78210127

schetlur-nv authored Nov 03, 2022



* Conditional dgrad/wgrad support
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>

* Fixing the change to depend only on requires_grad. Also updating LayerNorm MLP
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>

* Minor fixes.
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>

* Adding conditional wgrad for LayerNormLinear
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>

* bug fix and remove conditional dgrad

Co-authored-by: schetlur-nv schetlur@nvidia.com
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Adding unit test for wgrad disabled path
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>

* Adding more unit tests for wgrad disabled path
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>

* Adding unit tests for fp8 wgrad disabling, and cleaning up the code.
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>

* fix lint errors
Co-Authored-By: Sharan Chetlur <schetlur@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Co-authored-by: Sharan Chetlur <schetlur@dlcluster.nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

78210127

Update README.md (#27) · 924892fd

nzmora-nvidia authored Nov 03, 2022

Fix the sample code so it compiles after the signature of `te.Linear` has changed.
Signed-off-by: nzmora-nvidia <96238833+nzmora-nvidia@users.noreply.github.com>
Signed-off-by: nzmora-nvidia <96238833+nzmora-nvidia@users.noreply.github.com>

924892fd

31 Oct, 2022 1 commit

Test build as GitHub action (#13) · 415caeb6

Przemyslaw Tredak authored Oct 31, 2022



* Build the wheel as GitHub action
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

* Change the sanity test
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

415caeb6

28 Oct, 2022 1 commit
- Change the Blossom CI workflow · 92e6ee1d
  Przemek Tredak authored Oct 21, 2022
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  92e6ee1d
20 Oct, 2022 2 commits
- Changed version to 0.3.0dev · b2cda56c
  Przemek Tredak authored Oct 20, 2022
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  b2cda56c
- Add link to the documentation archives in the docs (#14) · 98cb5ac7
  Przemyslaw Tredak authored Oct 20, 2022
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  98cb5ac7
12 Oct, 2022 1 commit

Remove fp8_out from the LN API (#8) · 6a2161bf

Przemyslaw Tredak authored Oct 12, 2022



* Remove fp8_out from LN API
Signed-off-by: Przemyslaw Tredak <ptredak@nvidia.com>

* fix LN test
Signed-off-by: Przemyslaw Tredak <ptredak@nvidia.com>

* Fixes
Signed-off-by: Przemyslaw Tredak <ptredak@nvidia.com>
Signed-off-by: Przemyslaw Tredak <ptredak@nvidia.com>
Co-authored-by: ksivamani <ksivamani@nvidia.com>

6a2161bf

10 Oct, 2022 1 commit

Add C++ lint as GitHub action (#10) · 62f93325

Przemyslaw Tredak authored Oct 10, 2022



Add lint test as GitHub action
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

62f93325

07 Oct, 2022 1 commit
- Add Blossom CI yml (#9) · b0562288
  Przemyslaw Tredak authored Oct 07, 2022
```
Add blossom-ci.yml
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  b0562288
05 Oct, 2022 2 commits

fix documentation for default FP8 recipe format (#6) · 1568c6ed

Kirthi Shankar Sivamani authored Oct 05, 2022


Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

1568c6ed

rm deprecated pytest-runner requirement from setup (#7) · 2db83763

Kirthi Shankar Sivamani authored Oct 05, 2022


Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

2db83763

04 Oct, 2022 4 commits

Improve check in world size; None is valid group (#4) · a6b5f96d

Kirthi Shankar Sivamani authored Oct 04, 2022


Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

a6b5f96d

fix TP case to rm redundant AR (#3) · 5612ba78

Kirthi Shankar Sivamani authored Oct 04, 2022


Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

5612ba78

fp8_autocast bug fix when switching from non-fp8 execution (#2) · a2caec1e

Kirthi Shankar Sivamani authored Oct 04, 2022


Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

a2caec1e

Added the link to the User Guide (#1) · 1531dc78

Przemyslaw Tredak authored Oct 03, 2022


Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

1531dc78

28 Sep, 2022 2 commits

Changed VERSION to 0.2.0dev · 6c0469be
Przemek Tredak authored Sep 27, 2022
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
6c0469be

Inital code drop · 996ea169

Przemek Tredak authored Sep 27, 2022


Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

996ea169