This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new ea8b69b DeviceType enums match dlpack (#8407)
ea8b69b is described below
commit ea8b69ba98664ef22310a617ca45bc1fe369cfc2
Author: Schell Carl Scivally <[email protected]>
AuthorDate: Fri Jul 16 13:38:41 2021 +1200
DeviceType enums match dlpack (#8407)
* DeviceType enums match dlpack
* intentionally left blank
---
rust/tvm-sys/src/device.rs | 16 ++++++++--------
1 file changed, 8 insertions(+), 8 deletions(-)
diff --git a/rust/tvm-sys/src/device.rs b/rust/tvm-sys/src/device.rs
index 1da64fd..1ebac09 100644
--- a/rust/tvm-sys/src/device.rs
+++ b/rust/tvm-sys/src/device.rs
@@ -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 {