diff --git a/mindspore/ccsrc/device/gpu/gpu_kernel_runtime.cc b/mindspore/ccsrc/device/gpu/gpu_kernel_runtime.cc index 6c658f12e8..e8fd7ca922 100644 --- a/mindspore/ccsrc/device/gpu/gpu_kernel_runtime.cc +++ b/mindspore/ccsrc/device/gpu/gpu_kernel_runtime.cc @@ -171,7 +171,7 @@ bool GPUKernelRuntime::Run(session::KernelGraph *graph) { const uint64_t kUSecondInSecond = 1000000; uint64_t cost = kUSecondInSecond * static_cast(end_time.tv_sec - start_time.tv_sec); cost += static_cast(end_time.tv_usec - start_time.tv_usec); - MS_LOG(DEBUG) << "kernel runtime run graph in " << cost << " us"; + MS_LOG(DEBUG) << "GPU kernel runtime run graph in " << cost << " us"; return ret; } diff --git a/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.cc b/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.cc index 8443e4799f..80206f309d 100644 --- a/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.cc +++ b/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.cc @@ -104,12 +104,12 @@ DeviceAddressPtr GPUMemCopyManager::UpdateSwapInQueue() { return device_address; } -bool GPUMemCopyManager::AllocHostPinnedMem(size_t size, void **addr) { +bool GPUMemCopyManager::AllocHostPinnedMem(size_t size, void **addr) const { auto alloc_size = CudaDriver::AllocHostPinnedMem(size, addr); return alloc_size == size; } -void GPUMemCopyManager::FreeHostPinnedMem(void *addr) { CudaDriver::FreeHostPinnedMem(addr); } +void GPUMemCopyManager::FreeHostPinnedMem(void *addr) const { CudaDriver::FreeHostPinnedMem(addr); } void GPUMemCopyManager::ClearSwapQueue() { CHECK_OP_RET_WITH_EXCEPT(SyncMemCopyStream(SwapKind::kDeviceToHost), "Failed to sync swap out stream"); diff --git a/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.h b/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.h index a7cd8d4d8f..36ff273015 100644 --- a/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.h +++ b/mindspore/ccsrc/device/gpu/gpu_memory_copy_manager.h @@ -48,9 +48,9 @@ class GPUMemCopyManager : public MemCopyManager { DeviceAddressPtr UpdateSwapInQueue() override; - bool AllocHostPinnedMem(size_t size, void **addr) override; + bool AllocHostPinnedMem(size_t size, void **addr) const override; - void FreeHostPinnedMem(void *addr) override; + void FreeHostPinnedMem(void *addr) const override; void ClearSwapQueue() override; @@ -61,7 +61,6 @@ class GPUMemCopyManager : public MemCopyManager { std::queue> swap_in_queue_; }; using GPUMemCopyManagerPtr = std::shared_ptr; - } // namespace gpu } // namespace device } // namespace mindspore diff --git a/mindspore/ccsrc/pre_activate/mem_reuse/mem_copy_manager.h b/mindspore/ccsrc/pre_activate/mem_reuse/mem_copy_manager.h index 49d1884a48..ea9947b41b 100644 --- a/mindspore/ccsrc/pre_activate/mem_reuse/mem_copy_manager.h +++ b/mindspore/ccsrc/pre_activate/mem_reuse/mem_copy_manager.h @@ -83,9 +83,9 @@ class MemCopyManager { virtual DeviceAddressPtr UpdateSwapInQueue() { return nullptr; } - virtual bool AllocHostPinnedMem(size_t size, void **addr) { return true; } + virtual bool AllocHostPinnedMem(size_t size, void **addr) const { return true; } - virtual void FreeHostPinnedMem(void *addr) {} + virtual void FreeHostPinnedMem(void *addr) const {} virtual void ClearSwapQueue() {} };