diff --git a/lib/refrepo/gpu_ref.rb b/lib/refrepo/gpu_ref.rb index 48cb3fad7ae4dd0218328febbecc870b1c997ea7..5d83f58dad7f655a202822b9c98c14a270002ffc 100644 --- a/lib/refrepo/gpu_ref.rb +++ b/lib/refrepo/gpu_ref.rb @@ -6,6 +6,7 @@ class GPURef "GeForce GTX 1080 Ti" => 3584, "Tesla P100-PCIE-16GB" => 3584, "Tesla V100-PCIE-32GB" => 5120, + "Tesla V100-SXM2-32GB" => 5120, "Tesla M2075" => 448, "GeForce GTX TITAN Black" => 2880, "GeForce GTX 980" => 2048, @@ -17,6 +18,7 @@ class GPURef "GeForce GTX 1080 Ti" => "GTX 1080 Ti", "Tesla P100-PCIE-16GB" => "Tesla P100", "Tesla V100-PCIE-32GB" => "Tesla V100", + "Tesla V100-SXM2-32GB" => "Tesla V100", "Tesla M2075" => "Tesla M2075", "GeForce GTX TITAN Black" => "Titan Black", "GeForce GTX 980" => "GTX 980",