1 files changed, 2 insertions, 6 deletions
diff --git a/tests/cancel_inference_test.cpp b/tests/cancel_inference_test.cpp
index 846a29f..7f3fe74 100644
--- a/tests/cancel_inference_test.cpp
+++ b/tests/cancel_inference_test.cpp
@@ -40,9 +40,7 @@ int64_t defaultTimeout = 60000000000;
 
 void testCancelInference(const Device &device) {
     try {
-        auto networkBuffer = std::make_shared<Buffer>(device, sizeof(networkModelData));
-        std::memcpy(networkBuffer->data(), networkModelData, sizeof(networkModelData));
-        auto network = std::make_shared<Network>(device, networkBuffer);
+        auto network = std::make_shared<Network>(device, networkModelData, sizeof(networkModelData));
 
         std::vector<std::shared_ptr<Buffer>> inputBuffers;
         std::vector<std::shared_ptr<Buffer>> outputBuffers;
@@ -79,9 +77,7 @@ void testCancelInference(const Device &device) {
 
 void testRejectInference(const Device &device) {
     try {
-        auto networkBuffer = std::make_shared<Buffer>(device, sizeof(networkModelData));
-        std::memcpy(networkBuffer->data(), networkModelData, sizeof(networkModelData));
-        auto network = std::make_shared<Network>(device, networkBuffer);
+        auto network = std::make_shared<Network>(device, networkModelData, sizeof(networkModelData));
 
         std::vector<std::shared_ptr<Buffer>> inputBuffers;
         std::vector<std::shared_ptr<Buffer>> outputBuffers;