cugraph
cugraph copied to clipboard
[improvement] API changes for ONNX export and torch.compile(), scatter export WAR
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.
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.
/ok to test