Skip to content

Commit

Permalink
Revert "fix: GPU scheduling issue (Netflix#1520)" (Netflix#1542)
Browse files Browse the repository at this point in the history
This reverts commit 2465915.
  • Loading branch information
savingoyal authored and emattia committed Sep 17, 2023
1 parent 4b3f24f commit 56a5890
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions metaflow/plugins/kubernetes/kubernetes_decorator.py
Original file line number Diff line number Diff line change
Expand Up @@ -270,11 +270,6 @@ def step_init(self, flow, graph, step, decos, environment, flow_datastore, logge
)
)

# Treat gpu=0 as non-requests, but rewrite this as None for simpler handling in Kubernetes job/Argo Workflow templates.
# This is important, as adding a gpu limit with a value of 0 will be preferred for gpu nodes by the scheduler.
if self.attributes["gpu"] is not None and int(self.attributes["gpu"]) == 0:
self.attributes["gpu"] = None

if self.attributes["tmpfs_size"]:
if not (
isinstance(self.attributes["tmpfs_size"], (int, unicode, basestring))
Expand Down

0 comments on commit 56a5890

Please sign in to comment.