Adding conv tests with explicit algo definition #9798
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
virtan
requested review from
anderspapitto,
apaszke,
colesbury,
dzhulgakov,
ebetica and
ezyang
as code owners
virtan
requested review from
Yangqing,
gchanan,
goldsborough,
houseroad,
jamesr66a,
smessmer,
soumith,
ssnl,
zdevito and
zou3519
as code owners
|
@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
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
5 participants
No description provided.