Changeset View
Changeset View
Standalone View
Standalone View
mlir/include/mlir/Conversion/GPUToNVVM/GPUToNVVMPass.h
| Show All 25 Lines | |||||
| /// Configure target to convert from the GPU dialect to NVVM. | /// Configure target to convert from the GPU dialect to NVVM. | ||||
| void configureGpuToNVVMConversionLegality(ConversionTarget &target); | void configureGpuToNVVMConversionLegality(ConversionTarget &target); | ||||
| /// Collect a set of patterns to convert from the GPU dialect to NVVM. | /// Collect a set of patterns to convert from the GPU dialect to NVVM. | ||||
| void populateGpuToNVVMConversionPatterns(LLVMTypeConverter &converter, | void populateGpuToNVVMConversionPatterns(LLVMTypeConverter &converter, | ||||
| RewritePatternSet &patterns); | RewritePatternSet &patterns); | ||||
| /// Collect a set of patterns to convert WMMA ops from GPU dialect to NVVM. | |||||
| void populateGpuWMMAToNVVMConversionPatterns(LLVMTypeConverter &converter, | |||||
| RewritePatternSet &patterns); | |||||
| /// Creates a pass that lowers GPU dialect operations to NVVM counterparts. The | /// Creates a pass that lowers GPU dialect operations to NVVM counterparts. The | ||||
| /// index bitwidth used for the lowering of the device side index computations | /// index bitwidth used for the lowering of the device side index computations | ||||
| /// is configurable. | /// is configurable. | ||||
| std::unique_ptr<OperationPass<gpu::GPUModuleOp>> createLowerGpuOpsToNVVMOpsPass( | std::unique_ptr<OperationPass<gpu::GPUModuleOp>> createLowerGpuOpsToNVVMOpsPass( | ||||
| unsigned indexBitwidth = kDeriveIndexBitwidthFromDataLayout); | unsigned indexBitwidth = kDeriveIndexBitwidthFromDataLayout); | ||||
| } // namespace mlir | } // namespace mlir | ||||
| #endif // MLIR_CONVERSION_GPUTONVVM_GPUTONVVMPASS_H_ | #endif // MLIR_CONVERSION_GPUTONVVM_GPUTONVVMPASS_H_ | ||||