#include "coco/IR/Op.h"
+#include "coco/IR/ObjectManager.h"
#include <gtest/gtest.h>
TEST(IR_OP_CONV2D, ker_update)
{
+ // Prepare a kernel object for testing
+ coco::PtrLink<coco::Bag, coco::BagInfo> bag_link;
+ coco::ObjectManager obj_mgr{&bag_link};
+
+ auto obj = obj_mgr.create(nncc::core::ADT::kernel::Shape{1, 1, 3, 3});
+
+ // Test 'Conv2D' class
const coco::Conv2D::Param param{};
coco::Conv2D op{param};
- coco::KernelObject obj{nncc::core::ADT::kernel::Shape{1, 1, 3, 3}};
-
- op.ker(&obj);
- ASSERT_EQ(op.ker(), &obj);
+ op.ker(obj);
+ ASSERT_EQ(op.ker(), obj);
}