diff --git a/gpuallocator/staticdgx_policies.go b/gpuallocator/staticdgx_policies.go index 034feed..8b5168e 100644 --- a/gpuallocator/staticdgx_policies.go +++ b/gpuallocator/staticdgx_policies.go @@ -30,7 +30,7 @@ func NewStaticDGX1Policy(gpuType GPUType) Policy { // NewStaticDGX2Policy creates a new StaticDGX2Policy. func NewStaticDGX2Policy() Policy { - return &staticDGX1VoltaPolicy{} + return &staticDGX2VoltaPolicy{} } // Allocate GPUs following the Static DGX-1 policy for Pascal GPUs.