//===- ROCDLDialect.cpp - ROCDL IR Ops and Dialect registration -----------===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// // // This file defines the types and operation details for the ROCDL IR dialect in // MLIR, and the LLVM IR dialect. It also registers the dialect. // // The ROCDL dialect only contains GPU specific additions on top of the general // LLVM dialect. // //===----------------------------------------------------------------------===// #include "mlir/Dialect/LLVMIR/ROCDLDialect.h" #include "mlir/Dialect/GPU/IR/CompilationInterfaces.h" #include "mlir/Dialect/LLVMIR/LLVMDialect.h" #include "mlir/IR/Builders.h" #include "mlir/IR/BuiltinTypes.h" #include "mlir/IR/DialectImplementation.h" #include "mlir/IR/MLIRContext.h" #include "mlir/IR/Operation.h" #include "llvm/ADT/TypeSwitch.h" #include "llvm/AsmParser/Parser.h" #include "llvm/IR/Attributes.h" #include "llvm/IR/Function.h" #include "llvm/IR/Type.h" #include "llvm/Support/SourceMgr.h" usingnamespacemlir; usingnamespaceROCDL; #include "mlir/Dialect/LLVMIR/ROCDLOpsDialect.cpp.inc" //===----------------------------------------------------------------------===// // Parsing for ROCDL ops //===----------------------------------------------------------------------===// // <operation> ::= // `llvm.amdgcn.raw.buffer.load.* %rsrc, %offset, %soffset, %aux // : result_type` ParseResult RawBufferLoadOp::parse(OpAsmParser &parser, OperationState &result) { … } void RawBufferLoadOp::print(OpAsmPrinter &p) { … } // <operation> ::= // `llvm.amdgcn.raw.buffer.store.* %vdata, %rsrc, %offset, // %soffset, %aux : result_type` ParseResult RawBufferStoreOp::parse(OpAsmParser &parser, OperationState &result) { … } void RawBufferStoreOp::print(OpAsmPrinter &p) { … } // <operation> ::= // `llvm.amdgcn.raw.buffer.atomic.fadd.* %vdata, %rsrc, %offset, // %soffset, %aux : result_type` ParseResult RawBufferAtomicFAddOp::parse(OpAsmParser &parser, OperationState &result) { … } void RawBufferAtomicFAddOp::print(mlir::OpAsmPrinter &p) { … } // <operation> ::= // `llvm.amdgcn.raw.buffer.atomic.fmax.* %vdata, %rsrc, %offset, // %soffset, %aux : result_type` ParseResult RawBufferAtomicFMaxOp::parse(OpAsmParser &parser, OperationState &result) { … } void RawBufferAtomicFMaxOp::print(mlir::OpAsmPrinter &p) { … } // <operation> ::= // `llvm.amdgcn.raw.buffer.atomic.smax.* %vdata, %rsrc, %offset, // %soffset, %aux : result_type` ParseResult RawBufferAtomicSMaxOp::parse(OpAsmParser &parser, OperationState &result) { … } void RawBufferAtomicSMaxOp::print(mlir::OpAsmPrinter &p) { … } // <operation> ::= // `llvm.amdgcn.raw.buffer.atomic.umin.* %vdata, %rsrc, %offset, // %soffset, %aux : result_type` ParseResult RawBufferAtomicUMinOp::parse(OpAsmParser &parser, OperationState &result) { … } void RawBufferAtomicUMinOp::print(mlir::OpAsmPrinter &p) { … } //===----------------------------------------------------------------------===// // ROCDLDialect initialization, type parsing, and registration. //===----------------------------------------------------------------------===// // TODO: This should be the llvm.rocdl dialect once this is supported. void ROCDLDialect::initialize() { … } LogicalResult ROCDLDialect::verifyOperationAttribute(Operation *op, NamedAttribute attr) { … } //===----------------------------------------------------------------------===// // ROCDL target attribute. //===----------------------------------------------------------------------===// LogicalResult ROCDLTargetAttr::verify(function_ref<InFlightDiagnostic()> emitError, int optLevel, StringRef triple, StringRef chip, StringRef features, StringRef abiVersion, DictionaryAttr flags, ArrayAttr files) { … } #define GET_OP_CLASSES #include "mlir/Dialect/LLVMIR/ROCDLOps.cpp.inc" #define GET_ATTRDEF_CLASSES #include "mlir/Dialect/LLVMIR/ROCDLOpsAttributes.cpp.inc"