|
|
|
@ -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<std::pair<DeviceAddressPtr, DeviceEvent>> swap_in_queue_;
|
|
|
|
|
};
|
|
|
|
|
using GPUMemCopyManagerPtr = std::shared_ptr<GPUMemCopyManager>;
|
|
|
|
|
|
|
|
|
|
} // namespace gpu
|
|
|
|
|
} // namespace device
|
|
|
|
|
} // namespace mindspore
|
|
|
|
|