From 24dbc5c85f83cf72207ceae0cee49f6b48eab852 Mon Sep 17 00:00:00 2001
From: RissyRan <ran.rissy@gmail.com>
Date: Thu, 28 Nov 2024 00:17:43 +0000
Subject: [PATCH] Install full xpk dependency

---
 xlml/utils/xpk.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/xlml/utils/xpk.py b/xlml/utils/xpk.py
index 19f5c6a1b..02709146d 100644
--- a/xlml/utils/xpk.py
+++ b/xlml/utils/xpk.py
@@ -84,6 +84,7 @@ def run_workload(
     cmds = [
         "set -xue",
         f"git clone https://github.com/google/xpk {tmpdir}/xpk",
+        f"cd {tmpdir}/xpk && make install && export PATH=$PATH:$PWD/bin",
     ]
     if accelerator_type == GpuVersion.XPK_H100_MEGA.value:
       workload_create_cmd += " --scheduler=gke.io/topology-aware-auto"