added trition patch for cpu fix

Signed-off-by: mritunjaymusale <[email protected]>

Author: mritunjaymusale

patches/rocm-6.1.1/aotriton/0001-aotriton-rocm-home-dir-change.patch

@@ -1,7 +1,7 @@
-From ae27d084d624d1505c803abf13de43513a8a7515 Mon Sep 17 00:00:00 2001
+From 8ee78241c7c18a6c1941d8fd096194f2908ed9a3 Mon Sep 17 00:00:00 2001
 From: Mika Laitio <[email protected]>
 Date: Fri, 10 May 2024 17:05:37 -0700
-Subject: [PATCH 1/5] aotriton rocm home dir change
+Subject: [PATCH 1/6] aotriton rocm home dir change
 
 - change the rocm sdk directory location
   on all possible places
@@ -98,5 +98,5 @@ index 8117e5d..e0f2a57 100644
          per_kernel_targets = []
          for k in triton_kernels:
 -- 
-2.41.0
+2.45.1
 

patches/rocm-6.1.1/aotriton/0002-aotriton-triton-rocm-home-path-update.patch

@@ -1,7 +1,7 @@
-From 17b889089291c11e108a3b7bb011ff6f32b3b99d Mon Sep 17 00:00:00 2001
+From 1183293bec9ce1cbcb579b5b47e9370eda34d9f2 Mon Sep 17 00:00:00 2001
 From: Mika Laitio <[email protected]>
 Date: Fri, 10 May 2024 17:06:25 -0700
-Subject: [PATCH 2/5] aotriton triton rocm home path update
+Subject: [PATCH 2/6] aotriton triton rocm home path update
 
 - change the location of rocm_home
   to triton subproject
@@ -150,5 +150,5 @@ index 60eaa30..c9df795 100755
 +cp $ROCM_HOME/llvm/bin/ld.lld $TRITON_ROCM_DIR/bin/
 
 -- 
-2.41.0
+2.45.1
 

patches/rocm-6.1.1/aotriton/0003-include-mutex-for-unique_lock-method.patch

@@ -1,7 +1,7 @@
-From 41a90751beb8dd59e60e4515ac56c7e3416b2a65 Mon Sep 17 00:00:00 2001
+From 82bf0251d5f91b453cc1701127d76507c6ef15bb Mon Sep 17 00:00:00 2001
 From: Mika Laitio <[email protected]>
 Date: Tue, 14 May 2024 21:15:22 -0700
-Subject: [PATCH 3/5] include mutex for unique_lock method
+Subject: [PATCH 3/6] include mutex for unique_lock method
 
 fixes following error from hipcc:
 
@@ -83,5 +83,5 @@ index 0f96477..84a38ce 100644
  #include <zstd.h>
  #endif
 -- 
-2.41.0
+2.45.1
 

patches/rocm-6.1.1/aotriton/0004-added-more-cmake-options.patch

@@ -1,7 +1,7 @@
-From 1f1e03e53be0afa7541d50b19faf232ded2f014a Mon Sep 17 00:00:00 2001
+From 5883905718624eabf52db566d4644c35af8190f0 Mon Sep 17 00:00:00 2001
 From: Mika Laitio <[email protected]>
 Date: Mon, 20 May 2024 11:17:52 -0700
-Subject: [PATCH 4/5] added more cmake options
+Subject: [PATCH 4/6] added more cmake options
 
 following cmake options can now be passed from
 aotriton to triton
@@ -128,5 +128,5 @@ index 55425ad..02a0ccd 100644
              cmake_args.append("-DLLVM_EXTERNAL_LIT=" + lit_dir)
          cmake_args.extend(thirdparty_cmake_args)
 -- 
-2.41.0
+2.45.1
 

patches/rocm-6.1.1/aotriton/0005-work-in-progress-to-add-more-gpus.patch

@@ -1,7 +1,7 @@
-From 66e26bf6896c4fcfd864cf14ad4563298595cdd9 Mon Sep 17 00:00:00 2001
+From 44414d45fefd4296f6f06718bcd49acd1033f610 Mon Sep 17 00:00:00 2001
 From: Mika Laitio <[email protected]>
 Date: Mon, 20 May 2024 11:20:19 -0700
-Subject: [PATCH 5/5] work in progress to add more gpus
+Subject: [PATCH 5/6] work in progress to add more gpus
 
 this patch does not yet work, gpu
 capabilities needs to be double checked at least.
@@ -95,5 +95,5 @@ index 86ef5dc..1c3430b 100644
 +    'gfx1035'   : 'GPU_ARCH_AMD_GFX1035',
  }
 -- 
-2.41.0
+2.45.1
 

patches/rocm-6.1.1/aotriton/0006-changed-the-line-which-allocates-twice-the-number-of.patch

@@ -0,0 +1,27 @@
+From f9405e96aa6b6f8259ef8a95fbd70c8a9fc0f457 Mon Sep 17 00:00:00 2001
+From: mritunjaymusale <[email protected]>
+Date: Sun, 2 Jun 2024 18:00:59 +0530
+Subject: [PATCH 6/6] 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 02a0ccd..166f974 100644
+--- a/third_party/triton/python/setup.py
++++ b/third_party/triton/python/setup.py
+@@ -297,7 +297,7 @@ class CMakeBuild(build_ext):
+             build_args += ["--", "/m"]
+         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.1
+