MCPcopy Create free account
hub / github.com/catboost/catboost / TestTemplateKernel

Function TestTemplateKernel

catboost/cuda/cuda_lib/ut/test_serialization.cpp:36–54  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

34Y_UNIT_TEST_SUITE(TSerializationTest) {
35 template <class T>
36 void TestTemplateKernel(T val2) {
37 using TKernel = TSampleKernel2<T>;
38 TKernel kernel;
39 kernel.Data1 = 11;
40 kernel.Data2 = val2;
41
42 using TCmd = TGpuKernelTask<TKernel>;
43 TCmd cmd(TKernel(kernel), 1);
44
45 auto serialized = TTaskSerializer::Serialize(cmd);
46 auto deserialized = TTaskSerializer::LoadCommand(serialized);
47 auto deserializedPtr = dynamic_cast<TCmd*>(deserialized.Get());
48
49 UNIT_ASSERT_UNEQUAL(deserializedPtr, nullptr);
50 UNIT_ASSERT_EQUAL(deserializedPtr->GetStreamId(), 1u);
51 UNIT_ASSERT_EQUAL(deserializedPtr->GetCommandType(), ECommandType::StreamKernel);
52 UNIT_ASSERT_EQUAL(deserializedPtr->GetKernel().Data1, 11);
53 UNIT_ASSERT_EQUAL(deserializedPtr->GetKernel().Data2, val2);
54 }
55
56 Y_UNIT_TEST(TestKernelSerialization) {
57 {

Callers

nothing calls this directly

Calls 4

GetCommandTypeMethod · 0.80
SerializeFunction · 0.50
GetMethod · 0.45
GetStreamIdMethod · 0.45

Tested by

no test coverage detected