github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[RUNTIME][INTERPRETER] now also override __str__ method for tensors (#2325)

Browse Source
This commit is contained in:
Philippe Tillet
2023-09-17 16:49:30 -07:00
committed by GitHub
parent e686b4d6d4
commit 894fa9e943
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/runtime/interpreter.py
View File

@@ -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):