Skip to content

Commit 8fc32f1

Browse files
committed
Remove render_local_images_yaml
This commit removes rendering a file to specify the image / pull policy as this is now handled in the dpu operator developer tools - Rely on dpu operator Makefile to specify hardcoded image names - No longer rely on manually pulling images to the IPU, instead use pull-policy always Signed-off-by: Salvatore Daniele <[email protected]>
1 parent dbf3709 commit 8fc32f1

File tree

1 file changed

+8
-29
lines changed

1 file changed

+8
-29
lines changed

extraConfigDpu.py

Lines changed: 8 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -198,56 +198,39 @@ def build_dpu_operator_images() -> str:
198198
registry = _ensure_local_registry_running(lh, delete_all=True)
199199
reglocal.local_trust(lh)
200200

201-
operator_image = f"{registry}/{OPERATOR_IMG}"
202-
daemon_image = f"{registry}/{DAEMON_IMG}"
203-
render_local_images_yaml(operator_image=operator_image, daemon_image=daemon_image, outfilename=f"{REPO_DIR}/config/dev/local-images-template.yaml")
204-
205201
lh.run_or_die(f"make -C {REPO_DIR} local-buildx")
206202
lh.run_or_die(f"make -C {REPO_DIR} local-pushx")
207203

208204
return registry
209205

210206

211-
def start_dpu_operator(h: host.Host, client: K8sClient, operator_image: str, daemon_image: str, repo_wipe: bool = False) -> None:
207+
def start_dpu_operator(h: host.Host, client: K8sClient, repo_wipe: bool = False) -> None:
212208
logger.info(f"Deploying dpu operator containers on {h.hostname()}")
213209
if repo_wipe:
214210
h.run(f"rm -rf {REPO_DIR}")
215211
h.run_or_die(f"git clone {DPU_OPERATOR_REPO}")
216-
render_local_images_yaml(operator_image=operator_image, daemon_image=daemon_image, outfilename="/tmp/dpu-local-images-template.yaml", pull_policy="IfNotPresent")
217-
h.copy_to("/tmp/dpu-local-images-template.yaml", f"{REPO_DIR}/config/dev/local-images-template.yaml")
218212

213+
REGISTRY = host.LocalHost().hostname()
219214
h.run("dnf install -y pip")
220215
h.run_or_die("pip install yq")
221216
ensure_go_installed(h)
222-
reglocal.local_trust(h)
223-
h.run_or_die(f"podman pull {operator_image}")
224-
h.run_or_die(f"podman pull {daemon_image}")
225217
if h.is_localhost():
226218
env = os.environ.copy()
227219
env["KUBECONFIG"] = client._kc
228-
env["REGISTRY"] = host.LocalHost().hostname()
220+
env["REGISTRY"] = REGISTRY
229221
h.run(f"make -C {REPO_DIR} undeploy", env=env)
230222
ret = h.run(f"make -C {REPO_DIR} local-deploy", env=env)
231223
if not ret.success():
232224
logger.error_and_exit("Failed to deploy dpu operator")
233225
else:
234-
h.run(f"cd {REPO_DIR} && export KUBECONFIG={client._kc} && make undeploy")
235-
h.run_or_die(f"cd {REPO_DIR} && export KUBECONFIG={client._kc} && make local-deploy")
226+
env_cmds = f"export KUBECONFIG={client._kc} && export REGISTRY={REGISTRY}"
227+
h.run(f"cd {REPO_DIR} && {env_cmds} && make undeploy")
228+
h.run_or_die(f"cd {REPO_DIR} && {env_cmds} && make local-deploy")
236229
logger.info("Waiting for all dpu operator pods to become ready")
237230
time.sleep(30)
238231
client.oc_run_or_die("wait --for=condition=Ready pod --all -n openshift-dpu-operator --timeout=5m")
239232

240233

241-
def render_local_images_yaml(operator_image: str, daemon_image: str, outfilename: str, pull_policy: str = "Always") -> None:
242-
with open('./manifests/dpu/local-images-template.yaml.j2') as f:
243-
j2_template = jinja2.Template(f.read())
244-
rendered = j2_template.render(operator_image=operator_image, daemon_image=daemon_image, pull_policy=pull_policy)
245-
logger.info(rendered)
246-
247-
with open(outfilename, "w") as outFile:
248-
outFile.write(rendered)
249-
250-
251234
def ExtraConfigDpu(cc: ClustersConfig, cfg: ExtraConfigArgs, futures: dict[str, Future[Optional[host.Result]]]) -> None:
252235
[f.result() for (_, f) in futures.items()]
253236
logger.info("Running post config step to start DPU operator on IPU")
@@ -265,8 +248,6 @@ def ExtraConfigDpu(cc: ClustersConfig, cfg: ExtraConfigArgs, futures: dict[str,
265248
logger.info("Will not rebuild dpu-operator images")
266249
registry = _ensure_local_registry_running(lh, delete_all=False)
267250

268-
operator_image = f"{registry}/{OPERATOR_IMG}"
269-
daemon_image = f"{registry}/{DAEMON_IMG}"
270251

271252
# Build and start vsp on DPU
272253
vendor_plugin = init_vendor_plugin(acc)
@@ -280,7 +261,7 @@ def ExtraConfigDpu(cc: ClustersConfig, cfg: ExtraConfigArgs, futures: dict[str,
280261
acc.run_or_die(cmd)
281262
vendor_plugin.build_and_start(acc, client, registry)
282263

283-
start_dpu_operator(acc, client, operator_image, daemon_image, repo_wipe=True)
264+
start_dpu_operator(acc, client, repo_wipe=True)
284265

285266
# Disable firewall to ensure host-side can reach dpu
286267
acc.run("systemctl stop firewalld")
@@ -320,8 +301,6 @@ def ExtraConfigDpuHost(cc: ClustersConfig, cfg: ExtraConfigArgs, futures: dict[s
320301
else:
321302
logger.info("Will not rebuild dpu-operator images")
322303
registry = _ensure_local_registry_running(lh, delete_all=False)
323-
operator_image = f"{registry}/{OPERATOR_IMG}"
324-
daemon_image = f"{registry}/{DAEMON_IMG}"
325304

326305
# Need to trust the registry in OCP / Microshift
327306
logger.info("Ensuring local registry is trusted in OCP")
@@ -331,7 +310,7 @@ def ExtraConfigDpuHost(cc: ClustersConfig, cfg: ExtraConfigArgs, futures: dict[s
331310
vendor_plugin = init_vendor_plugin(h)
332311
vendor_plugin.build_and_start(lh, client, registry)
333312

334-
start_dpu_operator(lh, client, operator_image, daemon_image)
313+
start_dpu_operator(lh, client)
335314

336315
def helper(h: host.Host, node: NodeConfig) -> Optional[host.Result]:
337316
# Temporary workaround, remove once 4.16 installations are working

0 commit comments

Comments
 (0)