From 034273726c25b261f48c7b4ff690d2d342976e1b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?=
 <58388001+SzymonOzog@users.noreply.github.com>
Date: Wed, 16 Aug 2023 06:44:42 +0200
Subject: [PATCH] pass _buf to program

---
 tinygrad/runtime/ops_triton.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tinygrad/runtime/ops_triton.py b/tinygrad/runtime/ops_triton.py
index 26f95b449d..ed40707916 100644
--- a/tinygrad/runtime/ops_triton.py
+++ b/tinygrad/runtime/ops_triton.py
@@ -38,7 +38,8 @@ class TritonProgram:
     self.program = cuda.module_from_buffer(self.program.encode('utf-8')).get_function(self.program.split(".visible .entry ")[1].split("(")[0])
 
   def __call__(self, global_size, local_size, *args, wait=False) -> Any:
-    self.program(*[x for x in args], block = tuple(local_size), grid = tuple(global_size))
+    
+    self.program(*[x._buf for x in args], block = tuple(local_size), grid = tuple(global_size))
 
 def uops_to_triton(function_name:str, uops:List[UOp]):
     kernel = []