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

[PYTHON][OPS] Fixed typo in einsum

Browse Source
This commit is contained in:
Philippe Tillet
2020-02-25 16:37:55 -05:00
committed by Philippe Tillet
parent 646c49f847
commit ea37ba5d35
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -247,7 +247,7 @@ __global__ void {name}(
acc += a @ b; acc += a @ b;
#ifdef MASK #ifdef MASK
uint32 bits[TM, TN, TB] = bitcast<uint32[TM,TN,TB]>(acc); uint32 bits[TM, TN, TB] = bitcast<uint32[TM,TN,TB]>(acc);
acc = bitcast<TYPE[TM, TN, TB]>(bits & MASK); acc = bitcast<float[TM, TN, TB]>(bits & MASK);
#endif #endif
checkk = k > TK; checkk = k > TK;