pytorch
pytorch copied to clipboard
Tensors and Dynamic neural networks in Python with strong GPU acceleration
### 📚 The doc issue Hi, could you share the GPU Ops lists based on ROCm platform or how can I get that, thank you :) ### Suggest a potential...
### 🐛 Describe the bug Unskip the following test for ROCM5.0: test_import_stats | test_peak_memory | (__main__.TestImportTime) ### Versions ROCM 5.0
### 🐛 Describe the bug Unskip test_grid_sample on ROCM 5.0 test_nn | test_Conv2d_naive_groups_cuda_float16 | (__main__.TestNNDeviceTypeCUDA) test_nn | test_grid_sample | (__main__.TestNN) It was disabled by: https://github.com/pytorch/pytorch/issues/54535 ### Versions ROCM 5.0
### 🐛 Describe the bug The following tests are passing in ROCM5.0: test_nn | test_softmax_cuda_float32 | (__main__.TestNNDeviceTypeCUDA) test_nn | test_softmax_results_cuda_float16 | (__main__.TestNNDeviceTypeCUDA) test_nn | test_softmax_results_cuda_float32 | (__main__.TestNNDeviceTypeCUDA) We need to...
### 🐛 Describe the bug We need to unskip the following tests in test_ops.py. test_ops | test_fn_grad_min_binary_cuda_float64 | (__main__.TestGradientsCUDA) test_ops | test_fn_gradgrad_eig_cuda_complex128 | (__main__.TestGradientsCUDA) These are passing in ROCM5.0 ###...
### 🐛 Describe the bug Need to find out why these passing tests are skipped in upstream CI for `test_unary_ufuncs.py` test_unary_ufuncs test_reference_numerics_hard_polygamma_polygamma_n_2_cuda_float16 (__main__.TestUnaryUfuncsCUDA) test_unary_ufuncs test_reference_numerics_hard_polygamma_polygamma_n_2_cuda_float17 (__main__.TestUnaryUfuncsCUDA) test_unary_ufuncs test_reference_numerics_hard_polygamma_polygamma_n_2_cuda_float18 (__main__.TestUnaryUfuncsCUDA) test_unary_ufuncs...
### 🐛 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...
### 🐛 Describe the bug test_cuda | test_autocast_rnn | (__main__.TestCuda) -- | -- | -- ### Versions ROCm version: 4.5.2 PyTorch commit: https://github.com/pytorch/pytorch/commit/4b4f652f799c732a7280052e44a803a4cf6084de March 17, 2022 Test Audit logs ROCm...
### 🐛 Describe the bug test_ops | test_forward_mode_AD_nn_functional_conv_transpose3d_cuda_float64 | (__main__.TestGradientsCUDA) -- | -- | -- ### Versions ROCm version: 4.5.2 PyTorch commit: https://github.com/pytorch/pytorch/commit/4b4f652f799c732a7280052e44a803a4cf6084de March 17, 2022 Test Audit logs ROCm...
### 🐛 Describe the bug test_cuda | test_fixed_cuda_assert_async | (__main__.TestCuda) -- | -- | -- ### Versions ROCm version: 4.5.2 PyTorch commit: https://github.com/pytorch/pytorch/commit/4b4f652f799c732a7280052e44a803a4cf6084de March 17, 2022 Test Audit logs ROCm...