[python] basic support for pytorch seems to be working

2019-09-05 01:32:21 -04:00
parent ed0f706005
commit 65133cdf33
2 changed files with 26 additions and 15 deletions
--- a/python/examples/dot.py
+++ b/python/examples/dot.py
@@ -34,6 +34,5 @@ def run_torch():
  b = th.randn(K, N).cuda()
  th_c = th.matmul(a, b)
  tr_c = triton.ops.dot(a, b)
-  print(c)

 run_torch()