ROCm/pytorch

PyTorch CudaGraph UT support on ROCm

dllehr-amd opened this issue ยท 0 comments

๐Ÿ› 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