mlir/test/mlir-tblgen/shard-op-defs.td

   1 // RUN: mlir-tblgen -gen-op-defs -op-shard-count=2 -I %S/../../include %s | FileCheck %s --check-prefix=DEFS
   2 // RUN: mlir-tblgen -gen-op-decls -op-shard-count=2 -I %S/../../include %s | FileCheck %s --check-prefix=DECLS
   3
   4 include "mlir/IR/OpBase.td"
   5
   6 def Test_Dialect : Dialect {
   7   let name = "test";
   8   let cppNamespace = "test";
   9 }
  10
  11 class Test_Op<string mnemonic, list<Trait> traits = []>
  12     : Op<Test_Dialect, mnemonic, traits>;
  13
  14 def OpA : Test_Op<"a">;
  15 def OpB : Test_Op<"b">;
  16 def OpC : Test_Op<"c">;
  17
  18 // DECLS: OpA
  19 // DECLS: OpB
  20 // DECLS: OpC
  21 // DECLS: registerTestDialectOperations(
  22 // DECLS: registerTestDialectOperations0(
  23 // DECLS: registerTestDialectOperations1(
  24
  25 // DEFS-LABEL: GET_OP_DEFS_0
  26 // DEFS: void test::registerTestDialectOperations(
  27 // DEFS: void test::registerTestDialectOperations0(
  28 // DEFS: OpAAdaptor
  29 // DEFS: OpBAdaptor
  30
  31 // DEFS-LABEL: GET_OP_DEFS_1
  32 // DEFS: void test::registerTestDialectOperations1(
  33 // DEFS: OpCAdaptor