Extend support for Apple Silicon

Author: brandonros

src/device.rs

@@ -28,6 +28,8 @@ const AMD_DEVICE_ON_APPLE_VENDOR_STRING: &str = "AMD";
 const AMD_DEVICE_ON_APPLE_VENDOR_ID: u32 = 0x1021d00;
 const NVIDIA_DEVICE_VENDOR_STRING: &str = "NVIDIA Corporation";
 const NVIDIA_DEVICE_VENDOR_ID: u32 = 0x10de;
+const APPLE_DEVICE_VENDOR_ID: u32 = 0x1027F00;
+const APPLE_DEVICE_VENDOR_STRING: &str = "Apple";
 
 // The owned CUDA contexts are stored globally. Each devives contains an unowned reference, so
 // that devices can be cloned.
@@ -180,6 +182,8 @@ pub enum Vendor {
     Intel,
     /// GPU by NVIDIA.
     Nvidia,
+    /// GPU by Apple.
+    Apple,
 }
 
 impl TryFrom<&str> for Vendor {
@@ -191,6 +195,7 @@ impl TryFrom<&str> for Vendor {
             AMD_DEVICE_ON_APPLE_VENDOR_STRING => Ok(Self::Amd),
             INTEL_DEVICE_VENDOR_STRING => Ok(Self::Intel),
             NVIDIA_DEVICE_VENDOR_STRING => Ok(Self::Nvidia),
+            APPLE_DEVICE_VENDOR_STRING => Ok(Self::Apple),
             _ => Err(GPUError::UnsupportedVendor(vendor.to_string())),
         }
     }
@@ -205,6 +210,7 @@ impl TryFrom<u32> for Vendor {
             AMD_DEVICE_ON_APPLE_VENDOR_ID => Ok(Self::Amd),
             INTEL_DEVICE_VENDOR_ID => Ok(Self::Intel),
             NVIDIA_DEVICE_VENDOR_ID => Ok(Self::Nvidia),
+            APPLE_DEVICE_VENDOR_ID => Ok(Self::Apple),
             _ => Err(GPUError::UnsupportedVendor(format!("0x{:x}", vendor))),
         }
     }
@@ -216,6 +222,7 @@ impl fmt::Display for Vendor {
             Self::Amd => AMD_DEVICE_VENDOR_STRING,
             Self::Intel => INTEL_DEVICE_VENDOR_STRING,
             Self::Nvidia => NVIDIA_DEVICE_VENDOR_STRING,
+            Self::Apple => APPLE_DEVICE_VENDOR_STRING,
         };
         write!(f, "{}", vendor)
     }