[RUNTIME][INTERPRETER] now also override __str__ method for tensors (t…

…riton-lang#2325)
BinFan · Sep 17, 2023 · 894fa9e · 894fa9e
1 parent e686b4d
commit 894fa9e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/triton/runtime/interpreter.py b/python/triton/runtime/interpreter.py
@@ -370,6 +370,8 @@ def _patch_lang_tensor(tensor, builder):
             patch_attr(tensor, name, member, builder)
     tensor.__index__ = lambda self: int(self.handle.data)
     tensor.__bool__ = lambda self: True
+    tensor.__str__ = lambda self: str(self.handle.data)
+    tensor.__getitem__ = lambda self, slices: self.handle.data.__getitem__(slices)
 
 
 def _patch_lang_core(lang, builder):