Test PR #1
Open
Test PR #1
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Signed-off-by: Przemek Tredak <[email protected]>
ptrendx
force-pushed
the
pr_action_lint
branch
from
3d0e86f to
cc46a72
Compare
ptrendx
force-pushed
the
main
branch
6 times, most recently
from
e4e2198 to
22ef367
Compare
ptrendx
pushed a commit
that referenced
this pull request
Jan 26, 2023
* Add ONNX export support for TE modules (#1) * Add TorchScript Operators * Add symbolic methods to ONNX exporter * Add tests for the ONNX export Signed-off-by: Asfiya Baig <[email protected]> * fixes for pylint tests Signed-off-by: Asfiya Baig <[email protected]> * fix pylint warning in softmax.py Signed-off-by: Asfiya Baig <[email protected]> * move FP8 ORT lib inside tests/ Signed-off-by: Asfiya Baig <[email protected]> * enable cross attention tests Signed-off-by: Asfiya Baig <[email protected]> * refactor code by @nzmora * Increase layernorm FP16 threshold * Normalize onnx file names: _ separates configs; - separates words in a single config * Add get_attn_mask_str and fix mask string * Add missing ONNX files * Moved generated ONNX files to tests/gen_onnx_models/ Signed-off-by: Asfiya Baig <[email protected]> * fix merge conflict changes Signed-off-by: Asfiya Baig <[email protected]> * fix Q/DQ scale input Signed-off-by: Asfiya Baig <[email protected]> * enable FP16 config when bias is disabled Signed-off-by: Asfiya Baig <[email protected]> * fix pylint check errors Signed-off-by: Asfiya Baig <[email protected]> * updates 1. remove List import for pylint failure 2. address comments: remove state tensors from GPU 3. address comments: Update reverse_map_dtype function and add to namespace Signed-off-by: Asfiya Baig <[email protected]> * minor fix: coding guidelines Signed-off-by: Asfiya Baig <[email protected]> * changes: 1. skip FP8 tests on non-hopper devices 2. minor fix for C++ lint check Signed-off-by: Asfiya Baig <[email protected]> * fix onnxruntime version Signed-off-by: Asfiya Baig <[email protected]> * minor fix: add space between code and comment Signed-off-by: Asfiya Baig <[email protected]> * changes 1. update copyrights 2. update path to ORT .so Signed-off-by: Asfiya Baig <[email protected]> * Apply suggestions from code review Co-authored-by: Kirthi Shankar Sivamani <[email protected]> Signed-off-by: asfiyab-nvidia <[email protected]> Signed-off-by: Asfiya Baig <[email protected]> Signed-off-by: asfiyab-nvidia <[email protected]> Co-authored-by: Kirthi Shankar Sivamani <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.