Update comment

pragupta · web-flow · commit 149bf4edef65 · 2025-08-22T18:07:50.000-04:00
diff --git a/.github/scripts/build_triton_wheel.py b/.github/scripts/build_triton_wheel.py
@@ -162,6 +162,8 @@ def build_triton(
                 cwd=triton_basedir,
             )
 
+        # For gpt-oss models, triton requires this extra triton_kernels wheel
+        # triton_kernels came after pytorch release/2.8
         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")))

Original file line number	Diff line number	Diff line change
`@@ -162,6 +162,8 @@ def build_triton(`
`162`	`162`	`cwd=triton_basedir,`
`163`	`163`	`)`
`164`	`164`
	`165`	`+ # For gpt-oss models, triton requires this extra triton_kernels wheel`
	`166`	`+ # triton_kernels came after pytorch release/2.8`
`165`	`167`	`triton_kernels_dir = Path(f"{triton_basedir}/python/triton_kernels")`
`166`	`168`	`check_call([sys.executable, "-m", "build", "--wheel"], cwd=triton_kernels_dir, env=env)`
`167`	`169`	`kernels_whl_path = next(iter((triton_kernels_dir / "dist").glob("*.whl")))`