Refine current codes to support multi-devices
Created by: QiJune
-
Refine cuDNN related operators:
-
rename
GPUPlace
toCUDAPlace
and introduceCUDNNPlace
derived fromCUDAPlace
https://github.com/PaddlePaddle/Paddle/pull/6518 -
introduce
CUDNNDeviceContext
derived fromCUDADeviceContext
https://github.com/PaddlePaddle/Paddle/pull/6585 - make current cuDNN related operators to be a OpKernel of corresponding operator. For example: conv_cudnn is actually a OpKernel of conv operator, it should not be a independent operator. https://github.com/PaddlePaddle/Paddle/pull/6660
-
rename
-
Refine the interface of
OperatorBase::Run
, takevector<Place>
as parameter instead ofDeviceContext
https://github.com/PaddlePaddle/Paddle/pull/6783 -
Implement a pool of
DeviceContext
in init stage https://github.com/PaddlePaddle/Paddle/pull/6537 - refine operator run logic. https://github.com/PaddlePaddle/Paddle/pull/6783