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

[BACKEND] Added support for scalars in LoadOp / StoreOp / ElementwiseOp (#814)

Browse Source 
Also fixed various errors that showed up in `test_core.py`, and added more TODOs for open (hopefully relatively minor) issues
This commit is contained in:
Philippe Tillet
2022-10-28 01:17:55 -07:00
committed by GitHub
parent 3685194456
commit ac0f6793cc
6 changed files with 269 additions and 419 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/triton/Dialect/TritonGPU/IR/Dialect.h
View File

@@ -21,7 +21,7 @@ namespace mlir {
namespace triton {
namespace gpu {
unsigned getElemsPerThread(Attribute layout, ArrayRef<int64_t> shape);
unsigned getElemsPerThread(Type type);
SmallVector<unsigned> getSizePerThread(Attribute layout);