From 41144f927fbe14992f2286dd0e6324b4723fbadf Mon Sep 17 00:00:00 2001
From: Michael Melesse <micmelesse@gmail.com>
Date: Mon, 17 Oct 2022 20:41:28 +0000
Subject: [PATCH] fix hip launch

---
 python/triton/compiler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/triton/compiler.py b/python/triton/compiler.py
index bf1a97482..cfbca1b38 100644
--- a/python/triton/compiler.py
+++ b/python/triton/compiler.py
@@ -986,7 +986,7 @@ static inline void gpuAssert(hipError_t code, const char *file, int line)
 void _launch(int gridX, int gridY, int gridZ, int num_warps, int shared_memory, hipStream_t stream, hipFunction_t function, {arg_decls}) {{
   void *params[] = {{ {', '.join(f"&arg{i}" for i in signature.keys() if i not in constants)} }};
   if(gridX*gridY*gridZ > 0){{
-    hipLaunchKernel(function, dim3(gridX, gridY, gridZ), dim3(32*num_warps, 1, 1), 0 ,0, stream);
+    hipModuleLaunchKernel(function, gridX, gridY, gridZ, 32*num_warps, 1, 1, shared_memory, stream, params, 0);
   }}
 }}