[rocm7.1_internal_testing] Add triton_kernels wheel generation

pragupta · pragupta · commit 50889dc6b743 · 2025-08-22T21:43:55.000Z
diff --git a/.github/scripts/build_triton_wheel.py b/.github/scripts/build_triton_wheel.py
@@ -162,6 +162,11 @@ def build_triton(
                 cwd=triton_basedir,
             )
 
+        triton_kernels_dir = Path(f"{triton_basedir}/python/triton_kernels")
+        check_call([sys.executable, "-m", "build", "--wheel"], cwd=triton_kernels_dir, env=env)
+        kernels_whl_path = next(iter((triton_kernels_dir / "dist").glob("*.whl")))
+        shutil.copy(kernels_whl_path, Path.cwd())
+
         return Path.cwd() / whl_path.name
 
 

Original file line number	Diff line number	Diff line change
`@@ -162,6 +162,11 @@ def build_triton(`
`162`	`162`	`cwd=triton_basedir,`
`163`	`163`	`)`
`164`	`164`
	`165`	`+ triton_kernels_dir = Path(f"{triton_basedir}/python/triton_kernels")`
	`166`	`+ check_call([sys.executable, "-m", "build", "--wheel"], cwd=triton_kernels_dir, env=env)`
	`167`	`+ kernels_whl_path = next(iter((triton_kernels_dir / "dist").glob("*.whl")))`
	`168`	`+ shutil.copy(kernels_whl_path, Path.cwd())`
	`169`	`+`
`165`	`170`	`return Path.cwd() / whl_path.name`
`166`	`171`
`167`	`172`