Function GetPinnedHostMemoryCase

oneflow/core/memory/memory_case_util.cpp:30–37

}

void GetPinnedHostMemoryCase(const MemoryCase& mem_case, MemoryCase* ret) {
  ret->set_device_type(DeviceType::kCPU);
  ret->set_device_id(0);
  if (!IsHostMem(mem_case)) {
    ret->set_pinned_device_type(mem_case.device_type());
    ret->set_pinned_device_id(mem_case.device_id());
  }
}

MemoryCase GetPinnedHostMemoryCase(const MemoryCase& mem_case) {
  MemoryCase ret;