mlir/test/Dialect/IRDL/variadics.irdl.mlir

   1 // RUN: mlir-opt %s | mlir-opt | FileCheck %s
   2
   3 // CHECK: irdl.dialect @testvar {
   4 irdl.dialect @testvar {
   5
   6   // CHECK-LABEL: irdl.operation @single_operand {
   7   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i32
   8   // CHECK-NEXT:    irdl.operands(%[[v0]])
   9   // CHECK-NEXT:  }
  10   irdl.operation @single_operand {
  11     %0 = irdl.is i32
  12     irdl.operands(single %0)
  13   }
  14
  15   // CHECK-LABEL: irdl.operation @var_operand {
  16   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i16
  17   // CHECK-NEXT:    %[[v1:[^ ]*]] = irdl.is i32
  18   // CHECK-NEXT:    %[[v2:[^ ]*]] = irdl.is i64
  19   // CHECK-NEXT:    irdl.operands(%[[v0]], variadic %[[v1]], %[[v2]])
  20   // CHECK-NEXT:  }
  21   irdl.operation @var_operand {
  22     %0 = irdl.is i16
  23     %1 = irdl.is i32
  24     %2 = irdl.is i64
  25     irdl.operands(%0, variadic %1, %2)
  26   }
  27
  28   // CHECK-LABEL: irdl.operation @opt_operand {
  29   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i16
  30   // CHECK-NEXT:    %[[v1:[^ ]*]] = irdl.is i32
  31   // CHECK-NEXT:    %[[v2:[^ ]*]] = irdl.is i64
  32   // CHECK-NEXT:    irdl.operands(%[[v0]], optional %[[v1]], %[[v2]])
  33   // CHECK-NEXT:  }
  34   irdl.operation @opt_operand {
  35     %0 = irdl.is i16
  36     %1 = irdl.is i32
  37     %2 = irdl.is i64
  38     irdl.operands(%0, optional %1, %2)
  39   }
  40
  41   // CHECK-LABEL: irdl.operation @var_and_opt_operand {
  42   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i16
  43   // CHECK-NEXT:    %[[v1:[^ ]*]] = irdl.is i32
  44   // CHECK-NEXT:    %[[v2:[^ ]*]] = irdl.is i64
  45   // CHECK-NEXT:    irdl.operands(variadic %[[v0]], optional %[[v1]], %[[v2]])
  46   // CHECK-NEXT:  }
  47   irdl.operation @var_and_opt_operand {
  48     %0 = irdl.is i16
  49     %1 = irdl.is i32
  50     %2 = irdl.is i64
  51     irdl.operands(variadic %0, optional %1, %2)
  52   }
  53
  54
  55   // CHECK-LABEL: irdl.operation @single_result {
  56   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i32
  57   // CHECK-NEXT:    irdl.results(%[[v0]])
  58   // CHECK-NEXT:  }
  59   irdl.operation @single_result {
  60     %0 = irdl.is i32
  61     irdl.results(single %0)
  62   }
  63
  64   // CHECK-LABEL: irdl.operation @var_result {
  65   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i16
  66   // CHECK-NEXT:    %[[v1:[^ ]*]] = irdl.is i32
  67   // CHECK-NEXT:    %[[v2:[^ ]*]] = irdl.is i64
  68   // CHECK-NEXT:    irdl.results(%[[v0]], variadic %[[v1]], %[[v2]])
  69   // CHECK-NEXT:  }
  70   irdl.operation @var_result {
  71     %0 = irdl.is i16
  72     %1 = irdl.is i32
  73     %2 = irdl.is i64
  74     irdl.results(%0, variadic %1, %2)
  75   }
  76
  77   // CHECK-LABEL: irdl.operation @opt_result {
  78   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i16
  79   // CHECK-NEXT:    %[[v1:[^ ]*]] = irdl.is i32
  80   // CHECK-NEXT:    %[[v2:[^ ]*]] = irdl.is i64
  81   // CHECK-NEXT:    irdl.results(%[[v0]], optional %[[v1]], %[[v2]])
  82   // CHECK-NEXT:  }
  83   irdl.operation @opt_result {
  84     %0 = irdl.is i16
  85     %1 = irdl.is i32
  86     %2 = irdl.is i64
  87     irdl.results(%0, optional %1, %2)
  88   }
  89
  90   // CHECK-LABEL: irdl.operation @var_and_opt_result {
  91   // CHECK-NEXT:    %[[v0:[^ ]*]] = irdl.is i16
  92   // CHECK-NEXT:    %[[v1:[^ ]*]] = irdl.is i32
  93   // CHECK-NEXT:    %[[v2:[^ ]*]] = irdl.is i64
  94   // CHECK-NEXT:    irdl.results(variadic %[[v0]], optional %[[v1]], %[[v2]])
  95   // CHECK-NEXT:  }
  96   irdl.operation @var_and_opt_result {
  97     %0 = irdl.is i16
  98     %1 = irdl.is i32
  99     %2 = irdl.is i64
 100     irdl.results(variadic %0, optional %1, %2)
 101   }
 102 }