projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 532a709) | patch
Use correct workspace alloc call in MIOpen conv operator (#15712)
authorashishfarmer <ashish.farmer@amd.com>
Tue, 8 Jan 2019 19:23:01 +0000 (11:23 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 8 Jan 2019 19:38:45 +0000 (11:38 -0800)
commitcc402d8fa157cbd995a42dea41e38836db20fbc4
tree475665387a0009c8d8037cab0d4857c5f4456bd1tree | snapshot
parent532a7097710d5be9813b16d3e98f6d54f687efebcommit | diff
Use correct workspace alloc call in MIOpen conv operator (#15712)

Summary:
This PR contains changes for:
1. Using memory alloc from HIPContext while allocating workspace for MIOpen conv and transpose_conv operators rather than direct HIP mem alloc
2. Minor cleanup and removing an unnecessary sync call from MIOpen conv op

Differential Revision: D13598894

Pulled By: bddppq

fbshipit-source-id: 44886161abdf91cd29c7c93b3e23620e1b09c7c9
caffe2/operators/hip/conv_op_miopen.hip diff | blob | history
caffe2/operators/hip/conv_transpose_op_miopen.hip diff | blob | history
Domain: Machine Learning / ML Framework;