diff --git a/mlir/include/mlir/Dialect/CMakeLists.txt b/mlir/include/mlir/Dialect/CMakeLists.txt index 1c4569ecfa5848..9788e24e4a1d91 100644 --- a/mlir/include/mlir/Dialect/CMakeLists.txt +++ b/mlir/include/mlir/Dialect/CMakeLists.txt @@ -21,6 +21,7 @@ add_subdirectory(Math) add_subdirectory(MemRef) add_subdirectory(Mesh) add_subdirectory(MLProgram) +add_subdirectory(MPI) add_subdirectory(NVGPU) add_subdirectory(OpenACC) add_subdirectory(OpenACCMPCommon) diff --git a/mlir/include/mlir/Dialect/MPI/CMakeLists.txt b/mlir/include/mlir/Dialect/MPI/CMakeLists.txt new file mode 100644 index 00000000000000..f33061b2d87cff --- /dev/null +++ b/mlir/include/mlir/Dialect/MPI/CMakeLists.txt @@ -0,0 +1 @@ +add_subdirectory(IR) diff --git a/mlir/include/mlir/Dialect/MPI/IR/CMakeLists.txt b/mlir/include/mlir/Dialect/MPI/IR/CMakeLists.txt new file mode 100644 index 00000000000000..dc4b7a9087e609 --- /dev/null +++ b/mlir/include/mlir/Dialect/MPI/IR/CMakeLists.txt @@ -0,0 +1,25 @@ +add_mlir_dialect(MPI mpi) +add_mlir_doc(MPI MPI Dialects/ -gen-dialect-doc) + +# Add MPI operations +set(LLVM_TARGET_DEFINITIONS MPIOps.td) +mlir_tablegen(MPIOps.h.inc -gen-op-decls) +mlir_tablegen(MPIOps.cpp.inc -gen-op-defs) +add_public_tablegen_target(MLIRMPIOpsIncGen) +add_dependencies(mlir-generic-headers MLIRMPIOpsIncGen) + +# Add MPI types +set(LLVM_TARGET_DEFINITIONS MPITypes.td) +mlir_tablegen(MPITypesGen.h.inc -gen-typedef-decls) +mlir_tablegen(MPITypesGen.cpp.inc -gen-typedef-defs) +add_public_tablegen_target(MLIRMPITypesIncGen) +add_dependencies(mlir-generic-headers MLIRMPITypesIncGen) + +# Add MPI attributes +set(LLVM_TARGET_DEFINITIONS MPI.td) +mlir_tablegen(MPIEnums.h.inc -gen-enum-decls) +mlir_tablegen(MPIEnums.cpp.inc -gen-enum-defs) +mlir_tablegen(MPIAttrDefs.h.inc -gen-attrdef-decls) +mlir_tablegen(MPIAttrDefs.cpp.inc -gen-attrdef-defs) +add_public_tablegen_target(MLIRMPIAttrsIncGen) +add_dependencies(mlir-generic-headers MLIRMPIAttrsIncGen) diff --git a/mlir/include/mlir/Dialect/MPI/IR/MPI.h b/mlir/include/mlir/Dialect/MPI/IR/MPI.h new file mode 100644 index 00000000000000..f06b911ce3fe31 --- /dev/null +++ b/mlir/include/mlir/Dialect/MPI/IR/MPI.h @@ -0,0 +1,33 @@ +//===- MPI.h - MPI dialect ----------------------------------------*- C++-*-==// +// +// 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 +// +//===----------------------------------------------------------------------===// +#ifndef MLIR_DIALECT_MPI_IR_MPI_H_ +#define MLIR_DIALECT_MPI_IR_MPI_H_ + +#include "mlir/Bytecode/BytecodeOpInterface.h" +#include "mlir/IR/Dialect.h" +#include "mlir/IR/OpDefinition.h" +#include "mlir/IR/OpImplementation.h" + +//===----------------------------------------------------------------------===// +// MPIDialect +//===----------------------------------------------------------------------===// + +#include "mlir/Dialect/MPI/IR/MPIDialect.h.inc" + +#define GET_TYPEDEF_CLASSES +#include "mlir/Dialect/MPI/IR/MPITypesGen.h.inc" + +#include "mlir/Dialect/MPI/IR/MPIEnums.h.inc" + +#define GET_ATTRDEF_CLASSES +#include "mlir/Dialect/MPI/IR/MPIAttrDefs.h.inc" + +#define GET_OP_CLASSES +#include "mlir/Dialect/MPI/IR/MPIOps.h.inc" + +#endif // MLIR_DIALECT_MPI_IR_MPI_H_ diff --git a/mlir/include/mlir/Dialect/MPI/IR/MPI.td b/mlir/include/mlir/Dialect/MPI/IR/MPI.td new file mode 100644 index 00000000000000..f109260cdac59f --- /dev/null +++ b/mlir/include/mlir/Dialect/MPI/IR/MPI.td @@ -0,0 +1,183 @@ +//===- MPIBase.td - Base defs for mpi dialect ---------------*- tablegen -*-==// +// +// 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 +// +//===----------------------------------------------------------------------===// + +#ifndef MLIR_DIALECT_MPI_IR_MPI +#define MLIR_DIALECT_MPI_IR_MPI + +include "mlir/IR/AttrTypeBase.td" +include "mlir/IR/OpBase.td" +include "mlir/IR/EnumAttr.td" + +def MPI_Dialect : Dialect { + let name = "mpi"; + let cppNamespace = "::mlir::mpi"; + let description = [{ + This dialect models the Message Passing Interface (MPI), version 4.0. It is + meant to serve as an interfacing dialect that is targeted by higher-level dialects. + The MPI dialect itself can be lowered to multiple MPI implementations and hide + differences in ABI. The dialect models the functions of the MPI specification as + close to 1:1 as possible while preserving SSA value semantics where it makes sense, + and uses `memref` types instead of bare pointers. + + This dialect is under active development, and while stability is an + eventual goal, it is not guaranteed at this juncture. Given the early state, + it is recommended to inquire further prior to using this dialect. + + For an in-depth documentation of the MPI library interface, please refer to official documentation + such as the [OpenMPI online documentation](https://www.open-mpi.org/doc/current/). + }]; + + let usePropertiesForAttributes = 1; + let useDefaultAttributePrinterParser = 1; + let useDefaultTypePrinterParser = 1; +} + +//===----------------------------------------------------------------------===// +// Error classes enum: +//===----------------------------------------------------------------------===// + + + +def MpiCodeSuccess : I32EnumAttrCase<"MPI_SUCCESS", 0, "MPI_SUCCESS">; +def MpiCodeErrAccess : I32EnumAttrCase<"MPI_ERR_ACCESS", 1, "MPI_ERR_ACCESS">; +def MpiCodeErrAmode : I32EnumAttrCase<"MPI_ERR_AMODE", 2, "MPI_ERR_AMODE">; +def MpiCodeErrArg : I32EnumAttrCase<"MPI_ERR_ARG", 3, "MPI_ERR_ARG">; +def MpiCodeErrAssert : I32EnumAttrCase<"MPI_ERR_ASSERT", 4, "MPI_ERR_ASSERT">; +def MpiCodeErrBadFile : I32EnumAttrCase<"MPI_ERR_BAD_FILE", 5, "MPI_ERR_BAD_FILE">; +def MpiCodeErrBase : I32EnumAttrCase<"MPI_ERR_BASE", 6, "MPI_ERR_BASE">; +def MpiCodeErrBuffer : I32EnumAttrCase<"MPI_ERR_BUFFER", 7, "MPI_ERR_BUFFER">; +def MpiCodeErrComm : I32EnumAttrCase<"MPI_ERR_COMM", 8, "MPI_ERR_COMM">; +def MpiCodeErrConversion : I32EnumAttrCase<"MPI_ERR_CONVERSION", 9, "MPI_ERR_CONVERSION">; +def MpiCodeErrCount : I32EnumAttrCase<"MPI_ERR_COUNT", 10, "MPI_ERR_COUNT">; +def MpiCodeErrDims : I32EnumAttrCase<"MPI_ERR_DIMS", 11, "MPI_ERR_DIMS">; +def MpiCodeErrDisp : I32EnumAttrCase<"MPI_ERR_DISP", 12, "MPI_ERR_DISP">; +def MpiCodeErrDupDatarep : I32EnumAttrCase<"MPI_ERR_DUP_DATAREP", 13, "MPI_ERR_DUP_DATAREP">; +def MpiCodeErrErrhandler : I32EnumAttrCase<"MPI_ERR_ERRHANDLER", 14, "MPI_ERR_ERRHANDLER">; +def MpiCodeErrFile : I32EnumAttrCase<"MPI_ERR_FILE", 15, "MPI_ERR_FILE">; +def MpiCodeErrFileExists : I32EnumAttrCase<"MPI_ERR_FILE_EXISTS", 16, "MPI_ERR_FILE_EXISTS">; +def MpiCodeErrFileInUse : I32EnumAttrCase<"MPI_ERR_FILE_IN_USE", 17, "MPI_ERR_FILE_IN_USE">; +def MpiCodeErrGroup : I32EnumAttrCase<"MPI_ERR_GROUP", 18, "MPI_ERR_GROUP">; +def MpiCodeErrInfo : I32EnumAttrCase<"MPI_ERR_INFO", 19, "MPI_ERR_INFO">; +def MpiCodeErrInfoKey : I32EnumAttrCase<"MPI_ERR_INFO_KEY", 20, "MPI_ERR_INFO_KEY">; +def MpiCodeErrInfoNokey : I32EnumAttrCase<"MPI_ERR_INFO_NOKEY", 21, "MPI_ERR_INFO_NOKEY">; +def MpiCodeErrInfoValue : I32EnumAttrCase<"MPI_ERR_INFO_VALUE", 22, "MPI_ERR_INFO_VALUE">; +def MpiCodeErrInStatus : I32EnumAttrCase<"MPI_ERR_IN_STATUS", 23, "MPI_ERR_IN_STATUS">; +def MpiCodeErrIntern : I32EnumAttrCase<"MPI_ERR_INTERN", 24, "MPI_ERR_INTERN">; +def MpiCodeErrIo : I32EnumAttrCase<"MPI_ERR_IO", 25, "MPI_ERR_IO">; +def MpiCodeErrKeyval : I32EnumAttrCase<"MPI_ERR_KEYVAL", 26, "MPI_ERR_KEYVAL">; +def MpiCodeErrLocktype : I32EnumAttrCase<"MPI_ERR_LOCKTYPE", 27, "MPI_ERR_LOCKTYPE">; +def MpiCodeErrName : I32EnumAttrCase<"MPI_ERR_NAME", 28, "MPI_ERR_NAME">; +def MpiCodeErrNoMem : I32EnumAttrCase<"MPI_ERR_NO_MEM", 29, "MPI_ERR_NO_MEM">; +def MpiCodeErrNoSpace : I32EnumAttrCase<"MPI_ERR_NO_SPACE", 30, "MPI_ERR_NO_SPACE">; +def MpiCodeErrNoSuchFile : I32EnumAttrCase<"MPI_ERR_NO_SUCH_FILE", 31, "MPI_ERR_NO_SUCH_FILE">; +def MpiCodeErrNotSame : I32EnumAttrCase<"MPI_ERR_NOT_SAME", 32, "MPI_ERR_NOT_SAME">; +def MpiCodeErrOp : I32EnumAttrCase<"MPI_ERR_OP", 33, "MPI_ERR_OP">; +def MpiCodeErrOther : I32EnumAttrCase<"MPI_ERR_OTHER", 34, "MPI_ERR_OTHER">; +def MpiCodeErrPending : I32EnumAttrCase<"MPI_ERR_PENDING", 35, "MPI_ERR_PENDING">; +def MpiCodeErrPort : I32EnumAttrCase<"MPI_ERR_PORT", 36, "MPI_ERR_PORT">; +def MpiCodeErrProcAborted : I32EnumAttrCase<"MPI_ERR_PROC_ABORTED", 37, "MPI_ERR_PROC_ABORTED">; +def MpiCodeErrQuota : I32EnumAttrCase<"MPI_ERR_QUOTA", 38, "MPI_ERR_QUOTA">; +def MpiCodeErrRank : I32EnumAttrCase<"MPI_ERR_RANK", 39, "MPI_ERR_RANK">; +def MpiCodeErrReadOnly : I32EnumAttrCase<"MPI_ERR_READ_ONLY", 40, "MPI_ERR_READ_ONLY">; +def MpiCodeErrRequest : I32EnumAttrCase<"MPI_ERR_REQUEST", 41, "MPI_ERR_REQUEST">; +def MpiCodeErrRmaAttach : I32EnumAttrCase<"MPI_ERR_RMA_ATTACH", 42, "MPI_ERR_RMA_ATTACH">; +def MpiCodeErrRmaConflict : I32EnumAttrCase<"MPI_ERR_RMA_CONFLICT", 43, "MPI_ERR_RMA_CONFLICT">; +def MpiCodeErrRmaFlavor : I32EnumAttrCase<"MPI_ERR_RMA_FLAVOR", 44, "MPI_ERR_RMA_FLAVOR">; +def MpiCodeErrRmaRange : I32EnumAttrCase<"MPI_ERR_RMA_RANGE", 45, "MPI_ERR_RMA_RANGE">; +def MpiCodeErrRmaShared : I32EnumAttrCase<"MPI_ERR_RMA_SHARED", 46, "MPI_ERR_RMA_SHARED">; +def MpiCodeErrRmaSync : I32EnumAttrCase<"MPI_ERR_RMA_SYNC", 47, "MPI_ERR_RMA_SYNC">; +def MpiCodeErrRoot : I32EnumAttrCase<"MPI_ERR_ROOT", 48, "MPI_ERR_ROOT">; +def MpiCodeErrService : I32EnumAttrCase<"MPI_ERR_SERVICE", 49, "MPI_ERR_SERVICE">; +def MpiCodeErrSession : I32EnumAttrCase<"MPI_ERR_SESSION", 50, "MPI_ERR_SESSION">; +def MpiCodeErrSize : I32EnumAttrCase<"MPI_ERR_SIZE", 51, "MPI_ERR_SIZE">; +def MpiCodeErrSpawn : I32EnumAttrCase<"MPI_ERR_SPAWN", 52, "MPI_ERR_SPAWN">; +def MpiCodeErrTag : I32EnumAttrCase<"MPI_ERR_TAG", 53, "MPI_ERR_TAG">; +def MpiCodeErrTopology : I32EnumAttrCase<"MPI_ERR_TOPOLOGY", 54, "MPI_ERR_TOPOLOGY">; +def MpiCodeErrTruncate : I32EnumAttrCase<"MPI_ERR_TRUNCATE", 55, "MPI_ERR_TRUNCATE">; +def MpiCodeErrType : I32EnumAttrCase<"MPI_ERR_TYPE", 56, "MPI_ERR_TYPE">; +def MpiCodeErrUnknown : I32EnumAttrCase<"MPI_ERR_UNKNOWN", 57, "MPI_ERR_UNKNOWN">; +def MpiCodeErrUnsupportedDatarep : I32EnumAttrCase<"MPI_ERR_UNSUPPORTED_DATAREP", 58, "MPI_ERR_UNSUPPORTED_DATAREP">; +def MpiCodeErrUnsupportedOperation : I32EnumAttrCase<"MPI_ERR_UNSUPPORTED_OPERATION", 59, "MPI_ERR_UNSUPPORTED_OPERATION">; +def MpiCodeErrValueTooLarge : I32EnumAttrCase<"MPI_ERR_VALUE_TOO_LARGE", 60, "MPI_ERR_VALUE_TOO_LARGE">; +def MpiCodeErrWin : I32EnumAttrCase<"MPI_ERR_WIN", 61, "MPI_ERR_WIN">; +def MpiCodeErrLastcode : I32EnumAttrCase<"MPI_ERR_LASTCODE", 62, "MPI_ERR_LASTCODE">; + +def MpiErrorClassEnum : I32EnumAttr<"MpiErrorClassEnum", + "MPI error class name", + [ MpiCodeSuccess + ,MpiCodeErrAccess + ,MpiCodeErrAmode + ,MpiCodeErrArg + ,MpiCodeErrAssert + ,MpiCodeErrBadFile + ,MpiCodeErrBase + ,MpiCodeErrBuffer + ,MpiCodeErrComm + ,MpiCodeErrConversion + ,MpiCodeErrCount + ,MpiCodeErrDims + ,MpiCodeErrDisp + ,MpiCodeErrDupDatarep + ,MpiCodeErrErrhandler + ,MpiCodeErrFile + ,MpiCodeErrFileExists + ,MpiCodeErrFileInUse + ,MpiCodeErrGroup + ,MpiCodeErrInfo + ,MpiCodeErrInfoKey + ,MpiCodeErrInfoNokey + ,MpiCodeErrInfoValue + ,MpiCodeErrInStatus + ,MpiCodeErrIntern + ,MpiCodeErrIo + ,MpiCodeErrKeyval + ,MpiCodeErrLocktype + ,MpiCodeErrName + ,MpiCodeErrNoMem + ,MpiCodeErrNoSpace + ,MpiCodeErrNoSuchFile + ,MpiCodeErrNotSame + ,MpiCodeErrOp + ,MpiCodeErrOther + ,MpiCodeErrPending + ,MpiCodeErrPort + ,MpiCodeErrProcAborted + ,MpiCodeErrQuota + ,MpiCodeErrRank + ,MpiCodeErrReadOnly + ,MpiCodeErrRequest + ,MpiCodeErrRmaAttach + ,MpiCodeErrRmaConflict + ,MpiCodeErrRmaFlavor + ,MpiCodeErrRmaRange + ,MpiCodeErrRmaShared + ,MpiCodeErrRmaSync + ,MpiCodeErrRoot + ,MpiCodeErrService + ,MpiCodeErrSession + ,MpiCodeErrSize + ,MpiCodeErrSpawn + ,MpiCodeErrTag + ,MpiCodeErrTopology + ,MpiCodeErrTruncate + ,MpiCodeErrType + ,MpiCodeErrUnknown + ,MpiCodeErrUnsupportedDatarep + ,MpiCodeErrUnsupportedOperation + ,MpiCodeErrValueTooLarge + ,MpiCodeErrWin + ,MpiCodeErrLastcode]> { + let genSpecializedAttr = 0; + let cppNamespace = "::mlir::mpi"; +} + +def MpiErrorClassAttr : EnumAttr { + let assemblyFormat = "`<` $value `>`"; +} + +#endif // MLIR_DIALECT_MPI_IR_MPI diff --git a/mlir/include/mlir/Dialect/MPI/IR/MPIOps.td b/mlir/include/mlir/Dialect/MPI/IR/MPIOps.td new file mode 100644 index 00000000000000..1b96eb65a0d4bf --- /dev/null +++ b/mlir/include/mlir/Dialect/MPI/IR/MPIOps.td @@ -0,0 +1,196 @@ +//===- MPI.td - Message Passing Interface Ops --------------*- tablegen -*-===// +// +// 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 +// +//===----------------------------------------------------------------------===// + +#ifndef MPI_MLIR_IR_MPIOPS +#define MPI_MLIR_IR_MPIOPS + +include "mlir/Dialect/MPI/IR/MPI.td" +include "mlir/Dialect/MPI/IR/MPITypes.td" + +class MPI_Op traits = []> + : Op; + +//===----------------------------------------------------------------------===// +// InitOp +//===----------------------------------------------------------------------===// + +def MPI_InitOp : MPI_Op<"init", []> { + let summary = + "Initialize the MPI library, equivalent to `MPI_Init(NULL, NULL)`"; + let description = [{ + This operation must preceed most MPI calls (except for very few exceptions, + please consult with the MPI specification on these). + + Passing &argc, &argv is not supported currently. + + This operation can optionally return an `!mpi.retval` value that can be used + to check for errors. + }]; + + let results = (outs Optional:$retval); + + let assemblyFormat = "attr-dict (`:` type($retval)^)?"; +} + +//===----------------------------------------------------------------------===// +// CommRankOp +//===----------------------------------------------------------------------===// + +def MPI_CommRankOp : MPI_Op<"comm_rank", [ + +]> { + let summary = "Get the current rank, equivalent to " + "`MPI_Comm_rank(MPI_COMM_WORLD, &rank)`"; + let description = [{ + Communicators other than `MPI_COMM_WORLD` are not supprted for now. + + This operation can optionally return an `!mpi.retval` value that can be used + to check for errors. + }]; + + let results = ( + outs Optional : $retval, + I32 : $rank + ); + + let assemblyFormat = "attr-dict `:` type(results)"; +} + +//===----------------------------------------------------------------------===// +// SendOp +//===----------------------------------------------------------------------===// + +def MPI_SendOp : MPI_Op<"send", [ + +]> { + let summary = + "Equivalent to `MPI_Send(ptr, size, dtype, dest, tag, MPI_COMM_WORLD)`"; + let description = [{ + MPI_Send performs a blocking send of `size` elements of type `dtype` to rank `dest`. + The `tag` value and communicator enables the library to determine the matching of + multiple sends and receives between the same ranks. + + Communicators other than `MPI_COMM_WORLD` are not supprted for now. + + This operation can optionally return an `!mpi.retval` value that can be used + to check for errors. + }]; + + let arguments = (ins AnyMemRef : $ref, I32 : $tag, I32 : $rank); + + let results = (outs Optional:$retval); + + let assemblyFormat = "`(` $ref `,` $tag `,` $rank `)` attr-dict `:` " + "type($ref) `,` type($tag) `,` type($rank)" + "(`->` type($retval)^)?"; +} + +//===----------------------------------------------------------------------===// +// RecvOp +//===----------------------------------------------------------------------===// + +def MPI_RecvOp : MPI_Op<"recv", [ + +]> { + let summary = "Equivalent to `MPI_Recv(ptr, size, dtype, dest, tag, " + "MPI_COMM_WORLD, MPI_STATUS_IGNORE)`"; + let description = [{ + MPI_Recv performs a blocking receive of `size` elements of type `dtype` from rank `dest`. + The `tag` value and communicator enables the library to determine the matching of + multiple sends and receives between the same ranks. + + Communicators other than `MPI_COMM_WORLD` are not supprted for now. + The MPI_Status is set to `MPI_STATUS_IGNORE`, as the status object is not yet ported to MLIR. + + This operation can optionally return an `!mpi.retval` value that can be used + to check for errors. + }]; + + let arguments = (ins AnyMemRef : $ref, I32 : $tag, I32 : $rank); + + let results = (outs Optional:$retval); + + let assemblyFormat = "`(` $ref `,` $tag `,` $rank `)` attr-dict `:` " + "type($ref) `,` type($tag) `,` type($rank)" + "(`->` type($retval)^)?"; +} + + +//===----------------------------------------------------------------------===// +// FinalizeOp +//===----------------------------------------------------------------------===// + +def MPI_FinalizeOp : MPI_Op<"finalize", [ + +]> { + let summary = "Finalize the MPI library, equivalent to `MPI_Finalize()`"; + let description = [{ + This function cleans up the MPI state. Afterwards, no MPI methods may be invoked + (excpet for MPI_Get_version, MPI_Initialized, and MPI_Finalized). + Notably, MPI_Init cannot be called again in the same program. + + This operation can optionally return an `!mpi.retval` value that can be used + to check for errors. + }]; + + let results = (outs Optional:$retval); + + let assemblyFormat = "attr-dict (`:` type($retval)^)?"; +} + + +//===----------------------------------------------------------------------===// +// RetvalCheckOp +//===----------------------------------------------------------------------===// + +def MPI_RetvalCheckOp : MPI_Op<"retval_check", [ + +]> { + let summary = "Check an MPI return value against an error class"; + let description = [{ + + }]; + + let arguments = ( + ins MPI_Retval:$val, + MpiErrorClassAttr:$errclass + ); + + let results = ( + outs I1:$res + ); + + let assemblyFormat = "$val `=` $errclass attr-dict `:` type($res)"; +} + + + +//===----------------------------------------------------------------------===// +// RetvalCheckOp +//===----------------------------------------------------------------------===// + +def MPI_ErrorClassOp : MPI_Op<"error_class", [ + +]> { + let summary = "Get the error class from an error code, equivalent to the `MPI_Error_class` functoin."; + let description = [{ + + }]; + + let arguments = ( + ins MPI_Retval:$val + ); + + let results = ( + outs MPI_Retval:$errclass + ); + + let assemblyFormat = "$val attr-dict `:` type($val)"; +} + +#endif // MPI_MLIR_IR_MPIOPS diff --git a/mlir/include/mlir/Dialect/MPI/IR/MPITypes.td b/mlir/include/mlir/Dialect/MPI/IR/MPITypes.td new file mode 100644 index 00000000000000..1f1ae3144e6cea --- /dev/null +++ b/mlir/include/mlir/Dialect/MPI/IR/MPITypes.td @@ -0,0 +1,41 @@ +//===- MPITypes.td - Message Passing Interface types -------*- tablegen -*-===// +// +// 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 declares the Message Passing Interface dialect types. +// +//===----------------------------------------------------------------------===// + +#ifndef MLIR_DIALECT_MPI_IR_MPITYPES +#define MLIR_DIALECT_MPI_IR_MPITYPES + +include "mlir/IR/AttrTypeBase.td" +include "MPI.td" + +//===----------------------------------------------------------------------===// +// NPI Types +//===----------------------------------------------------------------------===// + +class MPI_Type traits = []> + : TypeDef { + let mnemonic = typeMnemonic; +} + +//===----------------------------------------------------------------------===// +// pdl::AttributeType +//===----------------------------------------------------------------------===// + +def MPI_Retval : MPI_Type<"Retval", "retval"> { + let summary = "MPI function call return value"; + let description = [{ + This type represents a value returned from an MPI call. This value can be + MPI_SUCCESS, MPI_ERR_IN_STATUS, or any error code. + }]; +} + + +#endif // MLIR_DIALECT_MPI_IR_MPITYPES \ No newline at end of file diff --git a/mlir/include/mlir/InitAllDialects.h b/mlir/include/mlir/InitAllDialects.h index 00f400aab5d50a..42087994d0f0c8 100644 --- a/mlir/include/mlir/InitAllDialects.h +++ b/mlir/include/mlir/InitAllDialects.h @@ -48,6 +48,7 @@ #include "mlir/Dialect/Linalg/Transforms/SubsetInsertionOpInterfaceImpl.h" #include "mlir/Dialect/Linalg/Transforms/TilingInterfaceImpl.h" #include "mlir/Dialect/MLProgram/IR/MLProgram.h" +#include "mlir/Dialect/MPI/IR/MPI.h" #include "mlir/Dialect/Math/IR/Math.h" #include "mlir/Dialect/MemRef/IR/MemRef.h" #include "mlir/Dialect/MemRef/IR/MemRefMemorySlot.h" @@ -120,6 +121,7 @@ inline void registerAllDialects(DialectRegistry ®istry) { memref::MemRefDialect, mesh::MeshDialect, ml_program::MLProgramDialect, + mpi::MPIDialect, nvgpu::NVGPUDialect, NVVM::NVVMDialect, omp::OpenMPDialect, diff --git a/mlir/lib/Dialect/CMakeLists.txt b/mlir/lib/Dialect/CMakeLists.txt index 68776a695cac4d..c72107939cf42b 100644 --- a/mlir/lib/Dialect/CMakeLists.txt +++ b/mlir/lib/Dialect/CMakeLists.txt @@ -21,6 +21,7 @@ add_subdirectory(Math) add_subdirectory(MemRef) add_subdirectory(Mesh) add_subdirectory(MLProgram) +add_subdirectory(MPI) add_subdirectory(NVGPU) add_subdirectory(OpenACC) add_subdirectory(OpenACCMPCommon) diff --git a/mlir/lib/Dialect/MPI/CMakeLists.txt b/mlir/lib/Dialect/MPI/CMakeLists.txt new file mode 100644 index 00000000000000..f33061b2d87cff --- /dev/null +++ b/mlir/lib/Dialect/MPI/CMakeLists.txt @@ -0,0 +1 @@ +add_subdirectory(IR) diff --git a/mlir/lib/Dialect/MPI/IR/CMakeLists.txt b/mlir/lib/Dialect/MPI/IR/CMakeLists.txt new file mode 100644 index 00000000000000..29d0b2379da747 --- /dev/null +++ b/mlir/lib/Dialect/MPI/IR/CMakeLists.txt @@ -0,0 +1,19 @@ +add_mlir_dialect_library(MLIRMPIDialect + MPIOps.cpp + MPI.cpp + + ADDITIONAL_HEADER_DIRS + ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/MPI + + DEPENDS + MLIRMPIIncGen + MLIRMPIOpsIncGen + MLIRMPITypesIncGen + MLIRMPIAttrsIncGen + + LINK_LIBS PUBLIC + MLIRDialect + MLIRIR + MLIRInferTypeOpInterface + MLIRSideEffectInterfaces + ) diff --git a/mlir/lib/Dialect/MPI/IR/MPI.cpp b/mlir/lib/Dialect/MPI/IR/MPI.cpp new file mode 100644 index 00000000000000..6c5f69febcd63d --- /dev/null +++ b/mlir/lib/Dialect/MPI/IR/MPI.cpp @@ -0,0 +1,56 @@ +//===- MPI.cpp - MPI dialect implementation -------------------------------===// +// +// 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 +// +//===----------------------------------------------------------------------===// + +#include "mlir/Dialect/MPI/IR/MPI.h" +#include "mlir/IR/Builders.h" +#include "mlir/IR/BuiltinAttributes.h" +#include "mlir/IR/DialectImplementation.h" +#include "llvm/ADT/TypeSwitch.h" + +using namespace mlir; +using namespace mlir::mpi; + +//===----------------------------------------------------------------------===// +/// Tablegen Definitions +//===----------------------------------------------------------------------===// + +#include "mlir/Dialect/MPI/IR/MPI.cpp.inc" + +#include "mlir/Dialect/MPI/IR/MPIDialect.cpp.inc" + +void MPIDialect::initialize() { + addOperations< +#define GET_OP_LIST +#include "mlir/Dialect/MPI/IR/MPIOps.cpp.inc" + >(); + + addTypes< +#define GET_TYPEDEF_LIST +#include "mlir/Dialect/MPI/IR/MPITypesGen.cpp.inc" + >(); + + addAttributes< +#define GET_ATTRDEF_LIST +#include "mlir/Dialect/MPI/IR/MPIAttrDefs.cpp.inc" + >(); +} + +//===----------------------------------------------------------------------===// +// TableGen'd dialect, type, and op definitions +//===----------------------------------------------------------------------===// + +#define GET_TYPEDEF_CLASSES +#include "mlir/Dialect/MPI/IR/MPITypesGen.cpp.inc" + +#include "mlir/Dialect/MPI/IR/MPIEnums.cpp.inc" + +#define GET_ATTRDEF_CLASSES +#include "mlir/Dialect/MPI/IR/MPIAttrDefs.cpp.inc" + +#define GET_OP_CLASSES +#include "mlir/Dialect/MPI/IR/MPIOps.cpp.inc" diff --git a/mlir/lib/Dialect/MPI/IR/MPIOps.cpp b/mlir/lib/Dialect/MPI/IR/MPIOps.cpp new file mode 100644 index 00000000000000..5827fabe664d4c --- /dev/null +++ b/mlir/lib/Dialect/MPI/IR/MPIOps.cpp @@ -0,0 +1,21 @@ +//===- MPIOps.cpp - MPI dialect ops implementation ------------===// +// +// 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 +// +//===----------------------------------------------------------------------===// + +#include "mlir/Dialect/MPI/IR/MPI.h" +#include "mlir/IR/Builders.h" +#include "mlir/IR/BuiltinAttributes.h" + +using namespace mlir; +using namespace mlir::mpi; + +//===----------------------------------------------------------------------===// +// TableGen'd op method definitions +//===----------------------------------------------------------------------===// + +#define GET_OP_CLASSES +#include "mlir/Dialect/MPI/IR/MPIOps.cpp.inc" diff --git a/mlir/test/Dialect/MPI/invalid.mlir b/mlir/test/Dialect/MPI/invalid.mlir new file mode 100644 index 00000000000000..1da154c7a58126 --- /dev/null +++ b/mlir/test/Dialect/MPI/invalid.mlir @@ -0,0 +1,50 @@ +// RUN: mlir-opt -split-input-file -verify-diagnostics %s + +// expected-error @+1 {{op result #0 must be 32-bit signless integer, but got 'i64'}} +%rank = mpi.comm_rank : i64 + +// ----- + +func.func @mpi_test(%ref : !llvm.ptr, %rank: i32) -> () { + // expected-error @+1 {{invalid kind of type specified}} + mpi.send(%ref, %rank, %rank) : !llvm.ptr, i32, i32 + + return +} + +// ----- + +func.func @mpi_test(%ref : !llvm.ptr, %rank: i32) -> () { + // expected-error @+1 {{invalid kind of type specified}} + mpi.recv(%ref, %rank, %rank) : !llvm.ptr, i32, i32 + + return +} + +// ----- + +func.func @mpi_test(%ref : memref<100xf32>, %rank: i32) -> () { + // expected-error @+1 {{'mpi.recv' op result #0 must be MPI function call return value, but got 'i32'}} + %res = mpi.recv(%ref, %rank, %rank) : memref<100xf32>, i32, i32 -> i32 + + return +} + +// ----- + +func.func @mpi_test(%ref : memref<100xf32>, %rank: i32) -> () { + // expected-error @+1 {{'mpi.send' op result #0 must be MPI function call return value, but got 'i32'}} + %res = mpi.send(%ref, %rank, %rank) : memref<100xf32>, i32, i32 -> i32 + + return +} + +// ----- + +func.func @mpi_test(%retval: !mpi.retval) -> () { + // expected-error @+2 {{custom op 'mpi.retval_check' expected ::mlir::mpi::MpiErrorClassEnum}} + // expected-error @+1 {{custom op 'mpi.retval_check' failed to parse MpiErrorClassAttr parameter 'value'}} + %res = mpi.retval_check %retval = + + return +} diff --git a/mlir/test/Dialect/MPI/ops.mlir b/mlir/test/Dialect/MPI/ops.mlir new file mode 100644 index 00000000000000..f7ffa23f7e82ea --- /dev/null +++ b/mlir/test/Dialect/MPI/ops.mlir @@ -0,0 +1,36 @@ +// RUN: mlir-opt %s | mlir-opt | FileCheck %s +// RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s + +func.func @mpi_test(%ref : memref<100xf32>) -> () { + // Note: the !mpi.retval result is optional on all operations except mpi.error_class + + // CHECK: %0 = mpi.init : !mpi.retval + %err = mpi.init : !mpi.retval + + // CHECK-NEXT: %retval, %rank = mpi.comm_rank : !mpi.retval, i32 + %retval, %rank = mpi.comm_rank : !mpi.retval, i32 + + // CHECK-NEXT: mpi.send(%arg0, %rank, %rank) : memref<100xf32>, i32, i32 + mpi.send(%ref, %rank, %rank) : memref<100xf32>, i32, i32 + + // CHECK-NEXT: %1 = mpi.send(%arg0, %rank, %rank) : memref<100xf32>, i32, i32 -> !mpi.retval + %err2 = mpi.send(%ref, %rank, %rank) : memref<100xf32>, i32, i32 -> !mpi.retval + + // CHECK-NEXT: mpi.recv(%arg0, %rank, %rank) : memref<100xf32>, i32, i32 + mpi.recv(%ref, %rank, %rank) : memref<100xf32>, i32, i32 + + // CHECK-NEXT: %2 = mpi.recv(%arg0, %rank, %rank) : memref<100xf32>, i32, i32 -> !mpi.retval + %err3 = mpi.recv(%ref, %rank, %rank) : memref<100xf32>, i32, i32 -> !mpi.retval + + // CHECK-NEXT: %3 = mpi.finalize : !mpi.retval + %rval = mpi.finalize : !mpi.retval + + // CHECK-NEXT: %4 = mpi.retval_check %retval = : i1 + %res = mpi.retval_check %retval = : i1 + + // CHECK-NEXT: %5 = mpi.error_class %0 : !mpi.retval + %errclass = mpi.error_class %err : !mpi.retval + + // CHECK-NEXT: return + func.return +}