Unverified Commit c4421f57 by Tianqi Chen Committed by GitHub

[SUBMODULE] update submodule to latest (#1728)

parent 8c5d3efa
Subproject commit 6f64f7866747a2a81bec84aea9bde0479c5b72c1
Subproject commit f519848d972c67971b4cbf8c34070d5a5e3ede0d
Subproject commit 10892ac964f1af7c81aae145cd3fab78bbccd297
Subproject commit bee4d1dd8dc1ee4a1fd8fa6a96476c2f8b7492a3
......@@ -62,11 +62,7 @@ typedef int64_t tvm_index_t;
typedef enum {
kDLAOCL = 5,
kDLSDAccel = 6,
kDLVulkan = 7,
kOpenGL = 11,
// Extension DRAM type, used for quickly test extension device
// The device api can differ depending on the xpu driver registered.
kExtDev = 12,
// AddExtraTVMType which is not in DLPack here
} TVMDeviceExtType;
......
......@@ -102,7 +102,7 @@ Target CreateTarget(const std::string& target_name,
} else if (target_name == "stackvm") {
t->device_type = kDLCPU;
} else if (target_name == "ext_dev") {
t->device_type = kExtDev;
t->device_type = kDLExtDev;
} else {
LOG(ERROR) << "Unknown target name " << target_name;
return target::stackvm();
......
......@@ -38,7 +38,7 @@ inline std::string DeviceName(int type) {
case kDLVPI: return "vpi";
case kDLROCM: return "rocm";
case kOpenGL: return "opengl";
case kExtDev: return "ext_dev";
case kDLExtDev: return "ext_dev";
default: LOG(FATAL) << "unknown type =" << type; return "Unknown";
}
}
......
......@@ -72,8 +72,7 @@ class VTADeviceAPI final : public DeviceAPI {
struct VTAWorkspacePool : public WorkspacePool {
VTAWorkspacePool() :
WorkspacePool(static_cast<DLDeviceType>(kExtDev),
VTADeviceAPI::Global()) {}
WorkspacePool(kDLExtDev, VTADeviceAPI::Global()) {}
};
void* VTADeviceAPI::AllocWorkspace(TVMContext ctx, size_t size, TVMType type_hint) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment