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

[PYTHON] Removed dead code for alloc_empty and register_scalar

Browse Source
This commit is contained in:
Philippe Tillet
2019-10-30 10:37:30 -04:00
parent f4fcaf84df
commit bf3dc63858
7 changed files with 3 additions and 172 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/examples/einsum.py
View File

@@ -24,7 +24,7 @@ cases = []
# Matmul
cases += [[[4, 1024, 1024], [1024, 1024], [4, 1024, 1024], "btc,ck->btk"]]
# Attention
cases += [[[4, 256, 8, 2, 64], [8, 2, 512, 64], [4, 256, 8, 2, 512], "bchak,hank->bchan"]]
# cases += [[[4, 256, 8, 2, 64], [8, 2, 512, 64], [4, 256, 8, 2, 512], "bchak,hank->bchan"]]
if mode == MODE.TF:
sess = tf.InteractiveSession()