Adding conv tests with explicit algo definition#9798
Adding conv tests with explicit algo definition#9798virtan wants to merge 25 commits intopytorch:masterfrom
Conversation
|
Thanks! This looks pretty good, cc @bddppq for a double check :) |
caffe2/python/pybind_state_gpu.cc
Outdated
| m.def("num_cuda_devices", &NumCudaDevices); | ||
| m.def("get_cuda_version", &CudaVersion); | ||
| m.def("get_cudnn_version", &cudnnCompiledVersion); | ||
| m.def("cudnn_convolution_fwd_algo_count", &CudnnConvolutionFwdAlgoCount); |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
| engine=st.sampled_from(["", "EIGEN"]), | ||
| shared_buffer=st.booleans(), | ||
| use_bias=st.booleans(), | ||
| force_algo_fwd=_cudnn_convolution_algo_count("fwd"), |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
| force_algo_dgrad=_cudnn_convolution_algo_count("dgrad"), | ||
| force_algo_wgrad=_cudnn_convolution_algo_count("wgrad"), | ||
| **hu.gcs) | ||
| def test_convolution_separate_stride_pad_layout(self, op_type, |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
| self.assertDeviceChecks(dc, op, inputs, [0]) | ||
| except RuntimeError as e: | ||
| es = str(e) | ||
| if es.find("status == CUDNN_STATUS_SUCCESS") < 0 \ |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
| except RuntimeError as e: | ||
| es = str(e) | ||
| if es.find("status == CUDNN_STATUS_SUCCESS") < 0 \ | ||
| or es.find("CUDNN_STATUS_NOT_SUPPORTED") < 0: |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
…o cudnn-conv-algo-tests
|
@pytorchbot retest this please |
|
I got into compilation issues, probably due to the merge with master. Re-merging again to confirm. |
|
Looks good, should actually pass tests |
facebook-github-bot
left a comment
There was a problem hiding this comment.
virtan has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Summary: Pull Request resolved: pytorch#9798 Differential Revision: D9034663 Pulled By: virtan fbshipit-source-id: d722f25f1dd00231ccc3ad5960bbbef63af02c2d
Summary: Pull Request resolved: pytorch#9798 Differential Revision: D9034663 Pulled By: virtan fbshipit-source-id: d722f25f1dd00231ccc3ad5960bbbef63af02c2d
No description provided.