Function MakeHostMemCase

oneflow/core/memory/memory_case_util.cpp:101–106

}

MemoryCase MakeHostMemCase() {
  MemoryCase mem_case;
  mem_case.set_device_type(DeviceType::kCPU);
  mem_case.set_device_id(0);
  return mem_case;
}

bool IsHostMem(const MemoryCase& mem_case) { return mem_case.device_type() == DeviceType::kCPU; }