diff options
| author | Mahesh Ravishankar <ravishankarm@google.com> | 2019-12-16 15:05:21 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-16 15:05:51 -0800 |
| commit | 80ec474a65a29a740b2edf7cc77d493ab4013a6b (patch) | |
| tree | ab816162cf4e1c228a33f6d6e1be9133072c05c3 /mlir/lib/Conversion/GPUToSPIRV | |
| parent | a0557ea9d6543a1be8451a59bd697cf01523607f (diff) | |
| download | bcm5719-llvm-80ec474a65a29a740b2edf7cc77d493ab4013a6b.tar.gz bcm5719-llvm-80ec474a65a29a740b2edf7cc77d493ab4013a6b.zip | |
Add atomic operations to SPIR-V dialect.
Some changes to the dialect generation script to allow specification
of different base class to derive from in ODS.
PiperOrigin-RevId: 285859230
Diffstat (limited to 'mlir/lib/Conversion/GPUToSPIRV')
| -rw-r--r-- | mlir/lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/mlir/lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.cpp b/mlir/lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.cpp index a8747a7a9bf..92cc02660a2 100644 --- a/mlir/lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.cpp +++ b/mlir/lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.cpp @@ -234,8 +234,9 @@ lowerAsEntryFunction(gpu::GPUFuncOp funcOp, SPIRVTypeConverter &typeConverter, "lowering as entry functions requires ABI info for all arguments"); return nullptr; } - // For entry functions need to make the signature void(void). Compute the - // replacement value for all arguments and replace all uses. + // Update the signature to valid SPIR-V types and add the ABI + // attributes. These will be "materialized" by using the + // LowerABIAttributesPass. TypeConverter::SignatureConversion signatureConverter(fnType.getNumInputs()); { for (auto argType : enumerate(funcOp.getType().getInputs())) { |
