Fix lce_ops_options_test.cc

larq · Jun 16, 2024 · 9dd6fa0 · 9dd6fa0
1 parent 9b3ac0c
commit 9dd6fa0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/larq_compute_engine/mlir/ir/lce_ops.h b/larq_compute_engine/mlir/ir/lce_ops.h
@@ -1,9 +1,9 @@
 #ifndef LARQ_COMPUTE_ENGINE_MLIR_IR_LCE_OPS_H_
 #define LARQ_COMPUTE_ENGINE_MLIR_IR_LCE_OPS_H_
 
+#include "mlir/Bytecode/BytecodeOpInterface.h"
 #include "mlir/Dialect/Quant/QuantTypes.h"
 #include "mlir/Interfaces/SideEffectInterfaces.h"
-#include "mlir/Bytecode/BytecodeOpInterface.h"
 
 // clang-format off
 #include "larq_compute_engine/mlir/ir/lce_dialect.h.inc"

diff --git a/larq_compute_engine/mlir/tests/lce_ops_options_test.cc b/larq_compute_engine/mlir/tests/lce_ops_options_test.cc
@@ -16,19 +16,19 @@ IntegerAttr getIntegerAttr(Builder builder, int value) {
 TEST(LCEOpsSerializationTest, QuantizeTest) {
   MLIRContext context;
   context.getOrLoadDialect<lq::LarqDialect>();
-  auto* op = Operation::create(
-      UnknownLoc::get(&context), OperationName("lq.Quantize", &context),
-      llvm::None, llvm::None, llvm::None, llvm::None, 0);
+  OperationState state(UnknownLoc::get(&context),
+                       OperationName("lq.Quantize", &context));
+  mlir::Operation* op = Operation::create(state);
 
   ASSERT_EQ(cast<lq::QuantizeOp>(op).buildCustomOptions().size(), 0);
 }
 
 TEST(LCEOpsSerializationTest, DequantizeTest) {
   MLIRContext context;
   context.getOrLoadDialect<lq::LarqDialect>();
-  auto* op = Operation::create(
-      UnknownLoc::get(&context), OperationName("lq.Dequantize", &context),
-      llvm::None, llvm::None, llvm::None, llvm::None, 0);
+  OperationState state(UnknownLoc::get(&context),
+                       OperationName("lq.Dequantize", &context));
+  mlir::Operation* op = Operation::create(state);
 
   ASSERT_EQ(cast<lq::DequantizeOp>(op).buildCustomOptions().size(), 0);
 }
@@ -37,9 +37,9 @@ TEST(LCEOpsSerializationTest, BConv2dTest) {
   MLIRContext context;
   context.getOrLoadDialect<lq::LarqDialect>();
   Builder builder(&context);
-  auto op = Operation::create(UnknownLoc::get(&context),
-                              OperationName("lq.Bconv2d", &context), llvm::None,
-                              llvm::None, llvm::None, llvm::None, 0);
+  OperationState state(UnknownLoc::get(&context),
+                       OperationName("lq.Bconv2d", &context));
+  mlir::Operation* op = Operation::create(state);
 
   op->setAttr("channels_in", getIntegerAttr(builder, 64));
   op->setAttr("dilation_height_factor", getIntegerAttr(builder, 3));
@@ -69,9 +69,9 @@ TEST(LCEOpsSerializationTest, BMaxPool2dTest) {
   MLIRContext context;
   context.getOrLoadDialect<lq::LarqDialect>();
   Builder builder(&context);
-  auto op = Operation::create(
-      UnknownLoc::get(&context), OperationName("lq.BMaxPool2d", &context),
-      llvm::None, llvm::None, llvm::None, llvm::None, 0);
+  OperationState state(UnknownLoc::get(&context),
+                       OperationName("lq.BMaxPool2d", &context));
+  mlir::Operation* op = Operation::create(state);
 
   op->setAttr("padding", builder.getStringAttr("SAME"));
   op->setAttr("stride_width", getIntegerAttr(builder, 2));