Skip to content

[PyTorch] Fix ONNX exports - #437

Merged
ksivaman merged 5 commits into
NVIDIA:mainfrom
ksivaman:fix_str_onnx_export
Sep 23, 2023
Merged

[PyTorch] Fix ONNX exports#437
ksivaman merged 5 commits into
NVIDIA:mainfrom
ksivaman:fix_str_onnx_export

Conversation

@ksivaman

Copy link
Copy Markdown
Member

A follow-up to #402 which broke the ONNX export due to dynamic string argument in fprop.

Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Comment threadtransformer_engine/pytorch/attention.py Outdated
Comment threadtransformer_engine/pytorch/attention.py Outdated
Comment threadtransformer_engine/pytorch/transformer.py Outdated
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
@ksivaman

Copy link
Copy Markdown
MemberAuthor

/te-ci

@ksivaman
ksivaman merged commit 903e1f4 into NVIDIA:mainSep 23, 2023
ptrendx pushed a commit that referenced this pull request Oct 2, 2023
* Fix ONNX exports
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* docs
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* review
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

@ksivaman@ptrendx