File tree 2 files changed +8
-8
lines changed
2 files changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -32,7 +32,7 @@ Eigen::DefaultDevice* CPUDeviceContext::eigen_device() const {
32
32
return eigen_device_.get ();
33
33
}
34
34
35
- Place CPUDeviceContext::place () const { return CPUPlace (); }
35
+ Place CPUDeviceContext::GetPlace () const { return CPUPlace (); }
36
36
37
37
#ifndef PADDLE_ONLY_CPU
38
38
@@ -50,7 +50,7 @@ CUDADeviceContext::CUDADeviceContext(GPUPlace place) : place_(place) {
50
50
51
51
CUDADeviceContext::~CUDADeviceContext () {
52
52
SetDeviceId (place_.device );
53
- wait ();
53
+ Wait ();
54
54
if (cublas_handle_) {
55
55
PADDLE_ENFORCE (dynload::cublasDestroy (cublas_handle_));
56
56
}
@@ -67,11 +67,11 @@ CUDADeviceContext::~CUDADeviceContext() {
67
67
PADDLE_ENFORCE (cudaStreamDestroy (stream_));
68
68
}
69
69
70
- Place CUDADeviceContext::place () const { return place_; }
70
+ Place CUDADeviceContext::GetPlace () const { return place_; }
71
71
72
72
cudaStream_t CUDADeviceContext::stream () const { return stream_; }
73
73
74
- void CUDADeviceContext::wait () const {
74
+ void CUDADeviceContext::Wait () const {
75
75
PADDLE_ENFORCE (cudaStreamSynchronize (stream_));
76
76
}
77
77
Original file line number Diff line number Diff line change @@ -31,7 +31,7 @@ namespace platform {
31
31
class DeviceContext {
32
32
public:
33
33
virtual ~DeviceContext () {}
34
- virtual Place place () const = 0;
34
+ virtual Place GetPlace () const = 0;
35
35
36
36
template <typename DeviceType>
37
37
DeviceType* get_eigen_device () const ;
@@ -45,7 +45,7 @@ class CPUDeviceContext : public DeviceContext {
45
45
46
46
Eigen::DefaultDevice* eigen_device () const ;
47
47
48
- Place place () const override ;
48
+ Place GetPlace () const override ;
49
49
50
50
private:
51
51
std::unique_ptr<Eigen::DefaultDevice> eigen_device_;
@@ -59,13 +59,13 @@ class CUDADeviceContext : public DeviceContext {
59
59
virtual ~CUDADeviceContext ();
60
60
61
61
/* ! \brief Wait for all operations completion in the stream. */
62
- void wait () const ;
62
+ void Wait () const ;
63
63
64
64
/* ! \brief Return CUDA stream in the device context. */
65
65
cudaStream_t stream () const ;
66
66
67
67
/* ! \brief Return place in the device context. */
68
- Place place () const override ;
68
+ Place GetPlace () const override ;
69
69
70
70
/* ! \brief Return eigen device in the device context. */
71
71
Eigen::GpuDevice* eigen_device () const ;
You can’t perform that action at this time.
0 commit comments