From 1ba5642f0a50fb59108341941eb240008dcff53b Mon Sep 17 00:00:00 2001 From: Ethan Wee <158101733+ethanwee1@users.noreply.github.com> Date: Thu, 17 Oct 2024 06:19:14 -0700 Subject: [PATCH 1/3] Update build_triton_wheel.py to update base repo name for triton --- .github/scripts/build_triton_wheel.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/scripts/build_triton_wheel.py b/.github/scripts/build_triton_wheel.py index 215f5f50d982a..27994d49f2771 100644 --- a/.github/scripts/build_triton_wheel.py +++ b/.github/scripts/build_triton_wheel.py @@ -95,10 +95,11 @@ def build_triton( with TemporaryDirectory() as tmpdir: triton_basedir = Path(tmpdir) / "triton" triton_pythondir = triton_basedir / "python" - triton_repo = "https://github.com/openai/triton" if build_rocm: + triton_repo = "https://github.com/ROCm/triton" triton_pkg_name = "pytorch-triton-rocm" else: + triton_repo = "https://github.com/openai/triton" triton_pkg_name = "pytorch-triton" check_call(["git", "clone", triton_repo], cwd=tmpdir) if release: From 5126f20b3b44250b4f4bef3677b34de2f8c61ebb Mon Sep 17 00:00:00 2001 From: Ethan Wee <158101733+ethanwee1@users.noreply.github.com> Date: Thu, 17 Oct 2024 10:14:56 -0700 Subject: [PATCH 2/3] Update build_triton_wheel.py to use ROCmSoftwarePlatform and not ROCm --- .github/scripts/build_triton_wheel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/scripts/build_triton_wheel.py b/.github/scripts/build_triton_wheel.py index 27994d49f2771..c9914202b4152 100644 --- a/.github/scripts/build_triton_wheel.py +++ b/.github/scripts/build_triton_wheel.py @@ -96,7 +96,7 @@ def build_triton( triton_basedir = Path(tmpdir) / "triton" triton_pythondir = triton_basedir / "python" if build_rocm: - triton_repo = "https://github.com/ROCm/triton" + triton_repo = "https://github.com/ROCmSoftwarePlatform/triton" triton_pkg_name = "pytorch-triton-rocm" else: triton_repo = "https://github.com/openai/triton" From 0b236e47a9777ea2508f7f961cd49ba655ef211c Mon Sep 17 00:00:00 2001 From: Ethan Wee <158101733+ethanwee1@users.noreply.github.com> Date: Fri, 18 Oct 2024 07:42:24 -0700 Subject: [PATCH 3/3] Update build_triton_wheel.py to use ROCm and not ROCmSoftwarePlatform --- .github/scripts/build_triton_wheel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/scripts/build_triton_wheel.py b/.github/scripts/build_triton_wheel.py index c9914202b4152..27994d49f2771 100644 --- a/.github/scripts/build_triton_wheel.py +++ b/.github/scripts/build_triton_wheel.py @@ -96,7 +96,7 @@ def build_triton( triton_basedir = Path(tmpdir) / "triton" triton_pythondir = triton_basedir / "python" if build_rocm: - triton_repo = "https://github.com/ROCmSoftwarePlatform/triton" + triton_repo = "https://github.com/ROCm/triton" triton_pkg_name = "pytorch-triton-rocm" else: triton_repo = "https://github.com/openai/triton"