feat: add Quadro RTX 6000

linxxx3 · Jun 16, 2020 · c529e1f · c529e1f
1 parent fae5e9f
commit c529e1f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -59,6 +59,7 @@ Currently only Nvidia hardware is supported, see [issue](https://github.com/fina
 
 ```
 ("Device_Name", Cores),
+("Quadro RTX 6000", 4608),
 ("TITAN RTX", 4608),
 ("Tesla V100", 5120),
 ("Tesla P100", 3584),

diff --git a/src/gpu/utils.rs b/src/gpu/utils.rs
@@ -26,6 +26,8 @@ pub fn get_devices(platform_name: &str) -> GPUResult<Vec<Device>> {
 lazy_static::lazy_static! {
     static ref CORE_COUNTS: HashMap<String, usize> = {
         let mut core_counts : HashMap<String, usize> = vec![
+            ("Quadro RTX 6000".to_string(), 4608),
+
             ("TITAN RTX".to_string(), 4608),
 
             ("Tesla V100".to_string(), 5120),