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

2022-12-06 09:57:05 -08:00
parent 532e10cf87
commit 115cd3ac47
1 changed files with 4 additions and 0 deletions
--- a/python/triton/language/core.py
+++ b/python/triton/language/core.py
@@ -758,6 +758,10 @@ def view(input, shape, _builder=None):
    shape = [x.value for x in shape]
    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