Skip to content

PyTorch CudaGraph UT support on ROCm #962

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.

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

Open
dllehr-amd opened this issue Mar 23, 2022 · 0 comments
Open

PyTorch CudaGraph UT support on ROCm #962

dllehr-amd opened this issue Mar 23, 2022 · 0 comments
Assignees
Labels
Unit Test Parity Parity with standard GPU

Comments

@dllehr-amd
Copy link

🐛 Describe the bug

Currently we do not have hipGraph support in PyTorch. We are working to add that implementation. In the meantime, the cudaGraph unit tests

test_cuda test_graph_capture_simple (main.TestCuda)
test_cuda test_graph_cudnn_dropout (main.TestCuda)
test_cuda test_graph_grad_scaling (main.TestCuda)
test_cuda test_graph_make_graphed_callables (main.TestCuda)
test_cuda test_graph_memory_stats_and_use_result_after_destroy_graph (main.TestCuda)
test_cuda test_graph_record_stream (main.TestCuda)
test_cuda test_graph_rng_distributions (main.TestCuda)
test_cuda test_graph_rng_functional (main.TestCuda)
test_cuda test_graph_three_successive (main.TestCuda)
test_cuda test_graph_two_successive (main.TestCuda)
test_cuda test_streaming_backwards_multiple_streams (main.TestCuda)

Will continue to be skipped.

Versions

ROCM 4.5.2

@dllehr-amd dllehr-amd added the Unit Test Parity Parity with standard GPU label Mar 23, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Unit Test Parity Parity with standard GPU
Projects
None yet
Development

No branches or pull requests

2 participants