forked from jiuyuan/InfiniTensor
31 lines
1.1 KiB
C++
31 lines
1.1 KiB
C++
#include "core/graph.h"
|
|
#include "core/runtime.h"
|
|
#include "operators/matmul.h"
|
|
#include "test.h"
|
|
|
|
namespace infini {
|
|
|
|
TEST(Hash, OperatorHash) {
|
|
OpPerfKey key1(0, OpType::Unknown), key2(0, OpType::Unknown);
|
|
{ // build with addOpWithOutputs
|
|
Graph g = make_ref<GraphObj>(nullptr);
|
|
Tensor i0 = g->addTensor({1, 2, 3}, DataType::UInt32);
|
|
Tensor w0 = g->addTensor({1, 3, 4}, DataType::UInt32);
|
|
Tensor o0 = g->addTensor({1, 2, 4}, DataType::UInt32);
|
|
auto matmul = g->addOpWithOutputs<MatmulObj>(i0, w0, o0);
|
|
key1 = matmul->getOpPerfKey();
|
|
EXPECT_NE(key1.hash, (HashType)0);
|
|
EXPECT_GT(key1.attrs.size(), (size_t)5);
|
|
}
|
|
{ // build with addOp
|
|
Graph g = make_ref<GraphObj>(nullptr);
|
|
Tensor i0 = g->addTensor({2, 2, 3}, DataType::UInt32);
|
|
Tensor w0 = g->addTensor({2, 3, 4}, DataType::UInt32);
|
|
auto matmul = g->addOp<MatmulObj>(i0, w0, nullptr);
|
|
key2 = matmul->getOpPerfKey();
|
|
EXPECT_NE(key2.hash, (HashType)0);
|
|
}
|
|
EXPECT_NE(key1.hash, key2.hash);
|
|
}
|
|
|
|
} // namespace infini
|