forked from OSchip/llvm-project
df71000d7d
This commit adds a pattern to lower linalg.generic for reduction to spv.GroupNonUniform* ops. Right now this only supports integer reduction on 1-D input memref. Shader entry point ABI is queried to make sure that the input memref's shape matches the local workgroup's invocation configuration. This makes sure that the workload fits in one local workgroup so that we can leverage SPIR-V group non-uniform operations. linglg.generic is a structured op that preserves the right level of information. It is easier to recognize reduction at this level than performing analysis on loops. This commit also exposes `getElementPtr` in SPIRVLowering.h given that it's a generally useful utility function. Differential Revision: https://reviews.llvm.org/D73437 |
||
|---|---|---|
| .. | ||
| Serialization | ||
| Transforms | ||
| CMakeLists.txt | ||
| DialectRegistration.cpp | ||
| LayoutUtils.cpp | ||
| SPIRVCanonicalization.td | ||
| SPIRVDialect.cpp | ||
| SPIRVLowering.cpp | ||
| SPIRVOps.cpp | ||
| SPIRVTypes.cpp | ||
| TargetAndABI.cpp | ||