Commits · 68adf451268ca9524ad556a9fe2206946db1a929 · OpenDAS / TransformerEngine

11 Nov, 2024 1 commit

Convert non-kernel cuda files to cpp (#1322) · 68adf451

Kirthi Shankar Sivamani authored Nov 11, 2024



* Fix file extensions
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



* upgrade paddle container for CI
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

68adf451

06 Nov, 2024 1 commit
- Update list of CI users (#1316) · 8f45c589
  Tim Moon authored Nov 06, 2024
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  8f45c589
25 Oct, 2024 1 commit

Support building documentation in Python 3.12 (#1274) · 8e4ee125

Tim Moon authored Oct 25, 2024



* Update dependencies for building documentation
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



* Patch RTD theme 3.0.0 to include version selector in sidebar
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Debug printing version in sidebar
Signed-off-by: Tim Moon <tmoon@nvidia.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

8e4ee125

24 Sep, 2024 1 commit
- Update list of CI users (#1203) · a44cb724
  Kirthi Shankar Sivamani authored Sep 24, 2024
```
Add new users to CI
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
  a44cb724
23 Sep, 2024 1 commit

Update list of CI users (#1198) · a68acd71

Tim Moon authored Sep 23, 2024



Add @pggPL to list of CI users
Signed-off-by: Tim Moon <tmoon@nvidia.com>

a68acd71

16 Sep, 2024 1 commit

Update CI users (#1181) · d2d4cf91

Tim Moon authored Sep 16, 2024



Update list of CI users
Signed-off-by: Tim Moon <tmoon@nvidia.com>

d2d4cf91

11 Sep, 2024 2 commits

Update CI users (#1180) · c55007b8
Tim Moon authored Sep 11, 2024
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
c55007b8

[PyTorch] Proxy class for low-precision tensor (#1127) · 2d57db8b

Tim Moon authored Sep 11, 2024



* Add base class for tensor proxies
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Move tensor detaching logic to tensor proxy base class
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Use Python wrappers to PyTorch extensions
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Include transpose caching logic in proxy encode function
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Debug dimension mismatch with amax history
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Move dequantize logic to proxy_decode func
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Rename to "QuantizedTensor"
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Rename "proxy_detach" to "detach"
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Include transpose cache in detach and clone funcs
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Fix linter warnings
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



* Update FP8 workspaces with QuantizedTensor functions
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Move logic for FP8 transpose cache in FP8 workspaces to base class
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Remove cast-transpose logic from linear op
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Remove unnecessary args for Float8Tensor when using FP8 attr dict
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Remove __torch_function__ to QuantizedTensor
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Fix linter warnings
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Update tests/pytorch/test_float8tensor.py
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

* Debug FP8 transpose test
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Debug cast functions
Signed-off-by: Tim Moon <tmoon@nvidia.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

2d57db8b

04 Sep, 2024 2 commits

Add user to TE CI (#1155) · 247850e8
Tim Moon authored Sep 03, 2024
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
247850e8

Bump actions/download-artifact from 3 to 4.1.7 in /.github/workflows (#1154) · af9f2fae

dependabot[bot] authored Sep 03, 2024

Bumps [actions/download-artifact](https://github.com/actions/download-artifact) from 3 to 4.1.7.
- [Release notes](https://github.com/actions/download-artifact/releases)
- [Commits](https://github.com/actions/download-artifact/compare/v3...v4.1.7

)

---
updated-dependencies:
- dependency-name: actions/download-artifact
  dependency-type: direct:production
...
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

af9f2fae

14 Aug, 2024 1 commit

Bump minimum CUDA version to 12.0 (#1103) · cc329b79

Tim Moon authored Aug 14, 2024



* Bump minimum CUDA version to 12.0
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Debug CUDA version check
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Debug CMake build
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Review suggestions from @ksivaman and @ptrendx

Remove logic for CUDA <12.0 in PyTorch and Paddle builds. Update version in docs and README.
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

cc329b79

13 Aug, 2024 1 commit

Use minimal CUDA container for PyTorch GitHub build (#1091) · dcc50c8e

Tim Moon authored Aug 13, 2024



* Use minimal CUDA container for PyTorch GitHub build
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Accidentally installed PyTorch in wrong test
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

* Debug sanity test
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

* Install PyTorch build dependencies
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

* Include NumPy as a dependency
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

* Disable sanity import test
Signed-off-by: Tim Moon <tmoon@nvidia.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

dcc50c8e

06 Aug, 2024 1 commit
- Add user to TE CI (#1081) · 6717554f
  Tim Moon authored Aug 06, 2024
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  6717554f
02 Aug, 2024 1 commit

Link attention docs to the main docs and fix errors reported by Sphinx (#1062) · 098e3006

Przemyslaw Tredak authored Aug 01, 2024



* Link attention docs to the main docs and fix errors reported by Sphinx
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

* Lower the version of nbsphinx
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

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

* Change the URL of example_attention.py to GitHub
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

* More fixes in the attention tutorial
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

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

098e3006

30 Jul, 2024 1 commit
- [Paddle] Update Paddle image (#1053) · 81dd6ad0
  Tian Zheng authored Jul 30, 2024
```
Update Paddle image
Signed-off-by: Tian Zheng <tizheng@nvidia.com>
```
  81dd6ad0
25 Jul, 2024 1 commit

Build scripts for pip wheels (#1036) · 09813578

Kirthi Shankar Sivamani authored Jul 24, 2024



* Specify python version
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Add classifiers for python
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Add utils to build wheels
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* make wheel scripts
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

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

* Fix paddle wheel
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* PaddlePaddle only builds for x86
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Add optional fwk deps
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Python3.8; catch install error
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* [wip] cudnn9 compile with paddle support
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* [wip] dont link cudnn
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

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

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

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

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

* dynamically load nvrtc
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



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

* remove residual packages; exclude stub from nvrtc .so search
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Exclude builtins from nvrtc .so search
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* properly include files for sdist
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* paddle wheel tie to python version
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Fix paddle build from src [wip]
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Fix workflow paddle build
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



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

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

* fix lint from pr986
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

* Add sanity wheel test
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Add sanity import to wheel test
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* remove upper limit on paddlepaddle version
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Remove unused imports
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Remove pybind11 dependency
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Fix cpp tests
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Search .sos in cuda home
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



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

* CLeanup, remove residual code
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

09813578

24 Jul, 2024 1 commit

Update minimum CMake version (#1037) · 9edcaf0e

Tim Moon authored Jul 24, 2024



* Set minimum CMake version to 3.21

Stop linking to nvtx.
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Update .github/workflows/build.yml
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

* Revert Python version to 3.9
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

9edcaf0e

16 Jul, 2024 1 commit
- Simplify logic for launching CI (#1001) · 210e57de
  Tim Moon authored Jul 16, 2024
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  210e57de
09 Jul, 2024 1 commit

Add test for building without support for any DL frameworks (#974) · 33dbf62b

Tim Moon authored Jul 09, 2024



* Add test for building with no DL frameworks
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Use NGC container for HPC SDK

NGC CUDA container does not have Git.
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Install dependencies in NGC CUDA container
Signed-off-by: Tim Moon <tmoon@nvidia.com>

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

* Debug sanity test
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Do sanity test out of repo
Signed-off-by: Tim Moon <tmoon@nvidia.com>

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

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

* Build with 4 parallel processes
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Maximize parallelism in GitHub builds
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Reduce number of PyTorch parallel build jobs
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Update docs/installation.rst
Co-authored-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>
Signed-off-by: Phuong Nguyen <36155692+phu0ngng@users.noreply.github.com>

* Disable parallel build for PyTorch test
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Signed-off-by: Phuong Nguyen <36155692+phu0ngng@users.noreply.github.com>
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>
Co-authored-by: Phuong Nguyen <36155692+phu0ngng@users.noreply.github.com>

33dbf62b

14 Jun, 2024 1 commit

Add auto-formatter (#919) · d99142a0

Kirthi Shankar Sivamani authored Jun 13, 2024



* Initial config test
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* remove linters, fix clang-format
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* fix clang-format
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* fix clang-format
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

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

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

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

* use config file
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

* pre-format fixes
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

* Add FA module
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

* Update CI configs
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

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

* revert accidental formatting changes
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* try with sudo
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

* fix pylint error properly
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* some review comments
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

* add fp8 attn include in the correct file
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

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

d99142a0

13 Jun, 2024 1 commit
- Add more type of change to the PR template (#924) · 6bec91e0
  Phuong Nguyen authored Jun 13, 2024
```
add more options for PR template
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
```
  6bec91e0
07 Jun, 2024 1 commit
- Fix minor security vulnerability when triggering CI (#898) · c6ce2b8c
  Tim Moon authored Jun 07, 2024
```
Fix bug when triggering CI
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  c6ce2b8c
29 May, 2024 1 commit

New NVIDIA footer in documentation (#876) · 4e30bc4b

Przemyslaw Tredak authored May 29, 2024



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

* Update docs toolchain versions
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

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

4e30bc4b

28 May, 2024 1 commit
- Add user to TE CI (#874) · 9bd938bc
  Tim Moon authored May 28, 2024
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  9bd938bc
01 May, 2024 1 commit

[JAX] Support FP8 training for Pipeline Parallelism when Micro-batch > 1 on Paxml. (#774) · eed4dfc6

Ming-Xu Huang authored May 01, 2024



* Support FP8 Meta Dtype (FM32) and Align FP8 Scale Update with PyTorch.
Signed-off-by: Ming Huang <mingh@nvidia.com>

* Modify with the feedback of code review
Signed-off-by: Ming Huang <mingh@nvidia.com>

* Hiding FlaxFloatMeta32 inside fp8.py
Signed-off-by: Ming Huang <mingh@nvidia.com>

* Make functions to be JAX tracable objects.
Signed-off-by: Ming Huang <mingh@nvidia.com>

* Rebased with mian.
Signed-off-by: Ming Huang <mingh@nvidia.com>

* Update jax images for github workflow.
Signed-off-by: Ming Huang <mingh@nvidia.com>

---------
Signed-off-by: Ming Huang <mingh@nvidia.com>

eed4dfc6

30 Apr, 2024 1 commit

Added pull request template (#793) · 816dd457

Przemyslaw Tredak authored Apr 30, 2024



* Added pull request template
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

* Changes from the review
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

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

816dd457

16 Apr, 2024 1 commit
- Add new users to TE CI · 1442b47e
  Przemek Tredak authored Apr 16, 2024
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  1442b47e
30 Jan, 2024 1 commit
- Add user to TE CI (#639) · ef4b1d1a
  Tim Moon authored Jan 30, 2024
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  ef4b1d1a
19 Jan, 2024 1 commit
- Add user to TE CI · a1e8f19d
  Przemek Tredak authored Jan 19, 2024
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  a1e8f19d
03 Jan, 2024 1 commit
- Change the copyright to include 2024 (#583) · cd798c97
  Przemyslaw Tredak authored Jan 02, 2024
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  cd798c97
13 Oct, 2023 2 commits
- Explicitly specify that PyTorch QA tests use PyTorch (#476) · 2c410836
  Tim Moon authored Oct 13, 2023
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  2c410836
- Remove remaining references to TensorFlow (#474) · 35e687d0
  Tim Moon authored Oct 13, 2023
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  35e687d0
11 Oct, 2023 1 commit
- Remove TF support (#467) · 1f4c3979
  Kirthi Shankar Sivamani authored Oct 11, 2023
```
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
  1f4c3979
09 Oct, 2023 1 commit
- Disable GitHub CI jobs using NGC PyTorch container (#462) · 79f5fac7
  Tim Moon authored Oct 09, 2023
```
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
  79f5fac7
06 Oct, 2023 1 commit
- Add user to TE CI and enable passing options to the CI (#460) · 38b85c35
  Przemyslaw Tredak authored Oct 06, 2023
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  38b85c35
04 Oct, 2023 1 commit
- Pin astroid to 2.15.7 (#457) · be67f219
  Przemyslaw Tredak authored Oct 04, 2023
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  be67f219
20 Jul, 2023 1 commit
- [GH Actions] Build PyTorch without FA for memory usage (#334) · 7a30ba45
  Kirthi Shankar Sivamani authored Jul 20, 2023
```
Build PyTorch with FA for memory usage
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
  7a30ba45
15 Jul, 2023 1 commit

Stop creating pip wheels in GitHub build tests (#315) · 36873ec8

Tim Moon authored Jul 15, 2023



* Stop creating pip wheels in GitHub build tests
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Revert to 23.03 container in GitHub build tests
Signed-off-by: Tim Moon <tmoon@nvidia.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

36873ec8

31 May, 2023 1 commit

Refactor build system (#235) · 37bbfc76

Tim Moon authored May 31, 2023



* Refactor Setuptools build system

Successfully launches CMake install, but installs CMake extensions in temp dir.
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Debug JAX build

Fix pybind11 import. Distinguish between build-time and run-time dependencies.
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Add helper function to determine dependencies
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Add missing license
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Debug case where system CMake is too old
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Add missing license
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Simplify sanity import tests

Just importing modules provides richer error messages.
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Properly install submodules
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Install helper library for TensorFlow
Signed-off-by: Tim Moon <tmoon@nvidia.com>

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

* Do not install Ninja by default
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Include Git commit hash in version string
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Override build_ext.build_extensions instead of build_ext.run
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Fix incorrect include path

Restore Ninja dependency. Restore overriding build_ext.run func.
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Review suggestions from @nouiz
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Disable parallel Ninja jobs in GitHub actions
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Properly install userbuffers lib
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Tweak install docs

Review suggestion from @ksivaman
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Add examples for specifying framework in docs
Signed-off-by: Tim Moon <tmoon@nvidia.com>

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

37bbfc76

11 May, 2023 1 commit
- Add users to CI · ce3980c8
  Przemek Tredak authored May 11, 2023
```
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>
```
  ce3980c8