aotriton, restore the cpu count patch from 6.1.1

fixes: #125

Signed-off-by: Mika Laitio <[email protected]>

patches/rocm-6.1.2/aotriton/0001-pass-extra-build-options.patch

@@ -1,7 +1,7 @@
-From 46c248cd100e89a6d16c3b889d4980b6c98c83a4 Mon Sep 17 00:00:00 2001
+From 5b47bd59cf54c9f3c259f6f870aac03c7add1a1d Mon Sep 17 00:00:00 2001
 From: Mika Laitio <[email protected]>
 Date: Mon, 29 Jul 2024 00:01:32 -0700
-Subject: [PATCH 1/2] pass extra build options
+Subject: [PATCH 1/3] pass extra build options
 
 Signed-off-by: Mika Laitio <[email protected]>
 ---

patches/rocm-6.1.2/aotriton/0002-add-gpus-with-gfx-name-to-build-list.patch

@@ -1,7 +1,7 @@
-From 2760a8141400c9573a0491d18bbcbaa1b0761bdd Mon Sep 17 00:00:00 2001
+From 1945003078824f6e885b667101dbcff61e56e3d0 Mon Sep 17 00:00:00 2001
 From: Mika Laitio <[email protected]>
 Date: Mon, 29 Jul 2024 12:48:47 -0700
-Subject: [PATCH 2/2] add gpus with gfx-name to build list
+Subject: [PATCH 2/3] add gpus with gfx-name to build list
 
 add all rocm sdk gpu's to build list
 and use the gfx* name for them instead

patches/rocm-6.1.2/aotriton/0003-changed-the-line-which-allocates-twice-the-number-of.patch

@@ -0,0 +1,27 @@
+From 11b43ea7a6b78f51a3fa44ac2054d185e16e89c3 Mon Sep 17 00:00:00 2001
+From: mritunjaymusale <[email protected]>
+Date: Sun, 2 Jun 2024 18:00:59 +0530
+Subject: [PATCH 3/3] changed the line which allocates twice the number of max
+ cpu threads to triton build
+
+Signed-off-by: mritunjaymusale <[email protected]>
+---
+ third_party/triton/python/setup.py | 2 +-
+ 1 file changed, 1 insertion(+), 1 deletion(-)
+
+diff --git a/third_party/triton/python/setup.py b/third_party/triton/python/setup.py
+index 390ee8b..4030cad 100644
+--- a/third_party/triton/python/setup.py
++++ b/third_party/triton/python/setup.py
+@@ -413,7 +413,7 @@ class CMakeBuild(build_ext):
+                 cmake_args += ["-A", "x64"]
+         else:
+             cmake_args += ["-DCMAKE_BUILD_TYPE=" + cfg]
+-            max_jobs = os.getenv("MAX_JOBS", str(2 * os.cpu_count()))
++            max_jobs = os.getenv("MAX_JOBS", str(os.cpu_count()))
+             build_args += ['-j' + max_jobs]
+
+         if check_env_flag("TRITON_BUILD_WITH_CLANG_LLD"):
+-- 
+2.45.2
+