diff --git a/lib/refrepo/gpu_ref.rb b/lib/refrepo/gpu_ref.rb index 36b735065ea067080943d0d0bc29e740d103328f..cfae48ba8cb52694b5c849181453e87b1a4053cc 100644 --- a/lib/refrepo/gpu_ref.rb +++ b/lib/refrepo/gpu_ref.rb @@ -5,6 +5,7 @@ class GPURef "GeForce RTX 2080 Ti" => 4352, "GeForce GTX 1080 Ti" => 3584, "Tesla P100-PCIE-16GB" => 3584, + "Tesla P100-SXM2-16GB" => 3584, "Tesla V100-PCIE-32GB" => 5120, "Tesla V100-SXM2-32GB" => 5120, "Tesla M2075" => 448, @@ -18,6 +19,7 @@ class GPURef "GeForce RTX 2080 Ti" => "RTX 2080 Ti", "GeForce GTX 1080 Ti" => "GTX 1080 Ti", "Tesla P100-PCIE-16GB" => "Tesla P100", + "Tesla P100-SXM2-16GB" => "Tesla P100", "Tesla V100-PCIE-32GB" => "Tesla V100", "Tesla V100-SXM2-32GB" => "Tesla V100", "Tesla M2075" => "Tesla M2075",