diff options
| author | Martin Polden <mpolden@mpolden.no> | 2023-09-08 09:00:27 +0200 |
|---|---|---|
| committer | Martin Polden <mpolden@mpolden.no> | 2023-09-08 09:00:27 +0200 |
| commit | 0161729355c19c63cfbc77d5f2a55f83de7be250 (patch) | |
| tree | e4b5ce2accccbd422e068103bf201fc74d67c079 /node-repository | |
| parent | f03c027f650362a8c2acea745ebc8e7526242b30 (diff) | |
Allow GPU in CD system
Diffstat (limited to 'node-repository')
| -rw-r--r-- | node-repository/src/main/java/com/yahoo/vespa/hosted/provision/provisioning/NodeRepositoryProvisioner.java | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/provisioning/NodeRepositoryProvisioner.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/provisioning/NodeRepositoryProvisioner.java index ac13ee992c2..f31ebcb0a47 100644 --- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/provisioning/NodeRepositoryProvisioner.java +++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/provisioning/NodeRepositoryProvisioner.java @@ -15,6 +15,7 @@ import com.yahoo.config.provision.NodeType; import com.yahoo.config.provision.ProvisionLock; import com.yahoo.config.provision.ProvisionLogger; import com.yahoo.config.provision.Provisioner; +import com.yahoo.config.provision.SystemName; import com.yahoo.config.provision.Zone; import com.yahoo.jdisc.Metric; import com.yahoo.transaction.Mutex; @@ -119,7 +120,7 @@ public class NodeRepositoryProvisioner implements Provisioner { if (!requested.minResources().nodeResources().gpuResources().equals(requested.maxResources().nodeResources().gpuResources())) throw new IllegalArgumentException(requested + " is invalid: GPU capacity cannot have ranges"); - if (!requested.minResources().nodeResources().gpuResources().isZero() && !zone.system().isPublic()) + if (!requested.minResources().nodeResources().gpuResources().isZero() && zone.system() == SystemName.main) throw new IllegalArgumentException(requested + " is invalid: GPUs are not supported in " + zone); logInsufficientDiskResources(cluster, requested, logger); |