gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[FRONTEND] Added reshape as an alias for view (for now) (#956)

Browse Source
This commit is contained in:
Philippe Tillet
2022-12-06 09:57:05 -08:00
committed by GitHub
parent 532e10cf87
commit 115cd3ac47
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/triton/language/core.py
View File

@@ -758,6 +758,10 @@ def view(input, shape, _builder=None):
shape = [x.value for x in shape] shape = [x.value for x in shape]
return semantic.view(input, shape, _builder) return semantic.view(input, shape, _builder)
@builtin
def reshape(input, shape, _builder=None):
# TODO: should be more than just a view
return view(input, shape, _builder)
# ----------------------- # -----------------------
# Linear Algebra # Linear Algebra