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

[CODEGEN][TRANSFORM][PEEPHOLE] Fixed bug in *1 multiplication

Browse Source
This commit is contained in:
Philippe Tillet
2020-02-18 23:04:46 -05:00
committed by Philippe Tillet
parent 9e54a03006
commit 4181f9f2af
3 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/ops/einsum.py
View File

@@ -321,7 +321,7 @@ __global__ void {name}(
}
"""
#print(src)
# print(src)
ret = triton.kernel(src)
if use_lut_a and lut_mode_a == _einsum.LUT_MODE.CONSTANT:
ret.set_constant('AD', delta_a)