Method get_tensor_device

imperative/python/src/dlpack_convertor.cpp:94–108

}

CompNode mgb::imperative::get_tensor_device(const DLDevice& ctx, int stream) {
    int id = ctx.device_id;
    switch (ctx.device_type) {
        case DLDeviceType::kDLCPU: {
            auto device = "cpu" + std::to_string(id);
            return as_comp_node(device);
        }
        case DLDeviceType::kDLCUDA: {
            auto device = "gpu" + std::to_string(id) + ":" + std::to_string(stream);
            return as_comp_node(device);
        }
        default:
            mgb_throw(MegBrainError, "Unsupported device_type");
    }
}

DType mgb::imperative::get_tensor_type(const DLDataType& dtype) {
    DType tensortype;