Skip to content

Conversation

@bddppq
Copy link
Contributor

@bddppq bddppq commented Jun 14, 2018

@petrex

This PR enable creating hip/miopen operators in Python.

$ python3.6 -c 'import caffe2.python._import_c_extension as C; print(C.has_hip_support); print(C.num_hip_devices())'
True
4

@bddppq bddppq requested review from Jorghi12, Yangqing and yinghai June 14, 2018 17:24
@petrex
Copy link
Contributor

petrex commented Jun 14, 2018

thanks @bddppq

@bddppq
Copy link
Contributor Author

bddppq commented Jun 14, 2018

@pytorchbot retest this please

@bddppq
Copy link
Contributor Author

bddppq commented Jun 15, 2018

pr/pytorch-win-ws2016-cuda9-cudnn7-py3 failure is unrelated, it's CI infra issue.

@bddppq bddppq merged commit a8bf30d into pytorch:master Jun 15, 2018
@bddppq bddppq deleted the hip-py branch June 15, 2018 02:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants