Use std::unique_ptr to create DeviceInterface

Signed-off-by: Dmitry Rogozhkin <[email protected]>

Author: dvrogozh

src/torchcodec/_core/DeviceInterface.cpp

@@ -36,7 +36,7 @@ bool registerDeviceInterface(
   return true;
 }
 
-std::shared_ptr<DeviceInterface> createDeviceInterface(
+std::unique_ptr<DeviceInterface> createDeviceInterface(
     const std::string device) {
   // TODO: remove once DeviceInterface for CPU is implemented
   if (device == "cpu") {
@@ -50,7 +50,7 @@ std::shared_ptr<DeviceInterface> createDeviceInterface(
       "Unsupported device: ",
       device);
 
-  return std::shared_ptr<DeviceInterface>(g_interface_map[deviceType](device));
+  return std::unique_ptr<DeviceInterface>(g_interface_map[deviceType](device));
 }
 
 } // namespace facebook::torchcodec

src/torchcodec/_core/DeviceInterface.h

@@ -59,7 +59,7 @@ bool registerDeviceInterface(
     const std::string deviceType,
     const CreateDeviceInterfaceFn createInterface);
 
-std::shared_ptr<DeviceInterface> createDeviceInterface(
+std::unique_ptr<DeviceInterface> createDeviceInterface(
     const std::string device);
 
 } // namespace facebook::torchcodec

test/VideoDecoderTest.cpp

@@ -5,6 +5,7 @@
 // LICENSE file in the root directory of this source tree.
 
 #include "src/torchcodec/_core/AVIOBytesContext.h"
+#include "src/torchcodec/_core/DeviceInterface.h"
 #include "src/torchcodec/_core/SingleStreamDecoder.h"
 
 #include <c10/util/Flags.h>