| 130 | } |
| 131 | |
| 132 | Maybe<DLDevice> ToDLDevice(Symbol<Device> ofdevice) { |
| 133 | DLDevice ctx; |
| 134 | ctx.device_id = ofdevice->device_id(); |
| 135 | switch (ofdevice->enum_type()) { |
| 136 | case DeviceType::kCPU: ctx.device_type = DLDeviceType::kDLCPU; break; |
| 137 | #ifdef WITH_CUDA |
| 138 | case DeviceType::kCUDA: ctx.device_type = DLDeviceType::kDLCUDA; break; |
| 139 | #endif |
| 140 | default: UNIMPLEMENTED_THEN_RETURN() << "Unsupported device type: " << ofdevice->type(); |
| 141 | } |
| 142 | return ctx; |
| 143 | } |
| 144 | |
| 145 | Maybe<DLDataType> ToDLDataType(DataType ofdtype) { |
| 146 | DLDataType dtype; |