tidl-viewer: fixes for yocto nativesdk build failures
[tidl/tidl-api.git] / examples / test / main.cpp
index 852cfe0549ec97f7c0554e1f07b496d7625eba78..645bb6fd07bdaa6f8f6e17318ba0fc55106bfd2a 100644 (file)
@@ -1,5 +1,5 @@
 /******************************************************************************
- * Copyright (c) 2017, Texas Instruments Incorporated - http://www.ti.com/
+ * Copyright (c) 2017-2018  Texas Instruments Incorporated - http://www.ti.com/
  *   All rights reserved.
  *
  *   Redistribution and use in source and binary forms, with or without
@@ -78,15 +78,14 @@ int main(int argc, char *argv[])
     signal(SIGTERM, exit);
 
     // If there are no devices capable of offloading TIDL on the SoC, exit
-    uint32_t num_dla =
-                Executor::GetNumDevicesSupportingTIDL(DeviceType::DLA);
-    uint32_t num_dsp =
-                Executor::GetNumDevicesSupportingTIDL(DeviceType::DSP);
+    uint32_t num_dla = Executor::GetNumDevices(DeviceType::DLA);
+    uint32_t num_dsp = Executor::GetNumDevices(DeviceType::DSP);
     if (num_dla == 0 && num_dsp == 0)
     {
         std::cout << "TI DL not supported on this SoC." << std::endl;
         return EXIT_SUCCESS;
     }
+    std::cout << "API Version: " << Executor::GetAPIVersion() << std::endl;
 
     // Process arguments
     std::string config_file;
@@ -151,10 +150,6 @@ bool RunConfiguration(const std::string& config_file, int num_devices,
     assert (input_data_file.good());
     assert (output_data_file.good());
 
-    // Determine input frame size from configuration
-    size_t frame_sz = configuration.inWidth * configuration.inHeight *
-                      configuration.inNumChannels;
-
     try
     {
         // Create a executor with the approriate core type, number of cores
@@ -170,8 +165,10 @@ bool RunConfiguration(const std::string& config_file, int num_devices,
         std::vector<void *> buffers;
         for (auto &eo : execution_objects)
         {
-            ArgInfo in  = { ArgInfo(malloc_ddr<char>(frame_sz), frame_sz)};
-            ArgInfo out = { ArgInfo(malloc_ddr<char>(frame_sz), frame_sz)};
+            size_t in_size  = eo->GetInputBufferSizeInBytes();
+            size_t out_size = eo->GetOutputBufferSizeInBytes();
+            ArgInfo in  = { ArgInfo(malloc(in_size),  in_size)};
+            ArgInfo out = { ArgInfo(malloc(out_size), out_size)};
             eo->SetInputOutputBuffer(in, out);
 
             buffers.push_back(in.ptr());
@@ -220,7 +217,7 @@ bool RunConfiguration(const std::string& config_file, int num_devices,
         }
 
         for (auto b : buffers)
-            __free_ddr(b);
+            free(b);
 
     }
     catch (tidl::Exception &e)
@@ -401,8 +398,8 @@ void ProcessArgs(int argc, char *argv[], std::string& config_file,
 
 void DisplayHelp()
 {
-    std::cout << "Usage: tidl\n"
-                 "  Will run all available networks if tidl is invoked without"
+    std::cout << "Usage: test_tidl\n"
+                 "  Will run all available networks if invoked without"
                  " any arguments.\n  Use -c to run a single network.\n"
                  "Optional arguments:\n"
                  " -c                   Path to the configuration file\n"