MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / compare_lite_tensor

Function compare_lite_tensor

lite/test/test_common.h:50–58  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

48
49template <typename T>
50void compare_lite_tensor(
51 std::shared_ptr<Tensor> tensor0, std::shared_ptr<Tensor> tensor1,
52 float maxerr = 1e-3) {
53 size_t elemsize = tensor0->get_layout().get_elem_size();
54 T* data_ptr0 = static_cast<T*>(tensor0->get_memory_ptr());
55 T* data_ptr1 = static_cast<T*>(tensor1->get_memory_ptr());
56 size_t length = tensor0->get_tensor_total_size_in_byte() / elemsize;
57 EXPECT_TRUE(compare_memory<T>(data_ptr0, data_ptr1, length, maxerr));
58}
59
60__attribute__((unused)) static std::shared_ptr<Tensor> get_input_data(
61 std::string path) {

Callers

nothing calls this directly

Calls 4

get_elem_sizeMethod · 0.80
get_layoutMethod · 0.45
get_memory_ptrMethod · 0.45

Tested by

no test coverage detected