[pytorch] clean-up of dynamic framework load

2019-09-05 02:16:27 -04:00
parent 65133cdf33
commit 44896ee777
5 changed files with 53 additions and 70 deletions
--- a/python/examples/dot.py
+++ b/python/examples/dot.py
@@ -34,5 +34,6 @@ def run_torch():
  b = th.randn(K, N).cuda()
  th_c = th.matmul(a, b)
  tr_c = triton.ops.dot(a, b)
+  print(tr_c)

 run_torch()