Skip to content

Commit

Permalink
DeviceType enums match dlpack (#8407)
Browse files Browse the repository at this point in the history
* DeviceType enums match dlpack

* intentionally left blank
  • Loading branch information
schell authored Jul 16, 2021
1 parent 63c6df8 commit ea8b69b
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions rust/tvm-sys/src/device.rs
Original file line number Diff line number Diff line change
Expand Up @@ -65,14 +65,14 @@ use thiserror::Error;
#[repr(i64)]
pub enum DeviceType {
CPU = 1,
CUDA,
CUDAHost,
OpenCL,
Vulkan,
Metal,
VPI,
ROCM,
ExtDev,
CUDA = 2,
CUDAHost = 3,
OpenCL = 4,
Vulkan = 7,
Metal = 8,
VPI = 9,
ROCM = 10,
ExtDev = 12,
}

impl Default for DeviceType {
Expand Down

0 comments on commit ea8b69b

Please sign in to comment.