Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[improvement] API changes for ONNX export and torch.compile(), scatter export WAR #4268

Closed
wants to merge 20 commits into from
Closed

[improvement] API changes for ONNX export and torch.compile(), scatter export WAR #4268

wants to merge 20 commits into from

Conversation

borisfom
Copy link

@borisfom borisfom commented Mar 21, 2024 •
edited
Loading

This PR depends on https://github.com/rapidsai/cugraph-ops/pull/624
We need to discuss/finalize proposed API changes.
Export tests added. Legacy ONNX, TRT and TS tests work, also compile().
dynamo_export and export.export() tests should probably be disabled for now.

@borisfom borisfom requested a review from a team as a code owner March 21, 2024 18:34
@copy-pr-bot copy-pr-bot
Copy link

copy-pr-bot bot commented Mar 21, 2024

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@borisfom borisfom changed the title Draft: API changes for ONNX export/tirch.compile(), scatter export WAR Draft: [improvement] API changes for ONNX export/tirch.compile(), scatter export WAR Mar 26, 2024
@borisfom borisfom changed the title Draft: [improvement] API changes for ONNX export/tirch.compile(), scatter export WAR [improvement] API changes for ONNX export/tirch.compile(), scatter export WAR Mar 26, 2024
@borisfom borisfom changed the title [improvement] API changes for ONNX export/tirch.compile(), scatter export WAR [improvement] API changes for ONNX export and torch.compile(), scatter export WAR Mar 26, 2024
@BradReesWork BradReesWork added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Apr 15, 2024
@BradReesWork
Copy link
Member

/ok to test

@BradReesWork BradReesWork added the DO NOT MERGE Hold off on merging; see PR for details label Apr 15, 2024
@rlratzel rlratzel changed the base branch from branch-24.06 to branch-24.08 May 24, 2024 19:04
@rlratzel rlratzel added this to the 24.08 milestone May 24, 2024
@borisfom borisfom requested review from a team as code owners May 24, 2024 22:15
@BradReesWork BradReesWork added DO NOT MERGE Hold off on merging; see PR for details and removed DO NOT MERGE Hold off on merging; see PR for details labels Jul 1, 2024
@BradReesWork BradReesWork modified the milestones: 24.08, 24.10 Aug 5, 2024
@BradReesWork BradReesWork changed the base branch from branch-24.08 to branch-24.10 August 5, 2024 19:20
@BradReesWork BradReesWork changed the base branch from branch-24.10 to branch-24.12 September 30, 2024 13:50
@BradReesWork BradReesWork modified the milestones: 24.10, 24.12 Sep 30, 2024
@BradReesWork
Copy link
Member

This PR is being closed since cugraph_equivariance has been moved out of this repo and into the BioNemo repp

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@tingyu66 tingyu66 Awaiting requested review from tingyu66

@wangxiaoyunNV wangxiaoyunNV Awaiting requested review from wangxiaoyunNV

Assignees
No one assigned
Labels
DO NOT MERGE Hold off on merging; see PR for details improvement Improvement / enhancement to an existing function non-breaking Non-breaking change python
Projects
None yet
Milestone
24.12
Development

Successfully merging this pull request may close these issues.
3 participants
@borisfom @BradReesWork @rlratzel