Merge pull request #7 from jackm321/tracing_fixes

jackm321 · web-flow · commit 2bcc4064c90e · 2019-03-14T13:42:05.000-07:00
Tracing fixes
diff --git a/foxi/onnxifi_dummy.c b/foxi/onnxifi_dummy.c
@@ -151,9 +151,10 @@ onnxReleaseGraph(onnxGraph graph) {
  * we have in this backend. It should be a subset of ALL_EXT_FUNCTION_LIST
  * in onnxifi_ext.h
  */
-const int extension_function_number = 2;
+const int extension_function_number = 3;
 const char* extension_function_list[] = {"onnxGetExtensionFunctionAddress",
-                                         "onnxSetIOAndRunGraph"};
+                                         "onnxSetIOAndRunGraph",
+                                         "onnxReleaseTraceEvents"};
 
 ONNXIFI_PUBLIC ONNXIFI_CHECK_RESULT onnxStatus ONNXIFI_ABI
 onnxGetExtensionFunctionAddress(
@@ -170,10 +171,14 @@ onnxGetExtensionFunctionAddress(
     if (strcmp(name, extension_function_list[i]) == 0) {
       switch (i) {
         case 0:
-          *function = &onnxGetExtensionFunctionAddress;
+          *function =
+            (onnxExtensionFunctionPointer)&onnxGetExtensionFunctionAddress;
           break;
         case 1:
-          *function = &onnxSetIOAndRunGraph;
+          *function = (onnxExtensionFunctionPointer)&onnxSetIOAndRunGraph;
+          break;
+        case 2:
+          *function = (onnxExtensionFunctionPointer)&onnxReleaseTraceEvents;
           break;
       }
     }
diff --git a/foxi/onnxifi_ext.h b/foxi/onnxifi_ext.h
@@ -111,7 +111,12 @@ typedef ONNXIFI_CHECK_RESULT onnxStatus
     const onnxTensorDescriptorV1* inputDescriptors,
     uint32_t outputsCount,
     const onnxTensorDescriptorV1* outputDescriptors,
-    onnxMemoryFenceV1* outputFence);
+    onnxMemoryFenceV1* outputFence,
+    onnxTraceEventList* traceEvents);
+    
+typedef ONNXIFI_CHECK_RESULT onnxStatus
+  (ONNXIFI_ABI* onnxReleaseTraceEventsFunction)(
+    onnxTraceEventList* traceEvents);
 
 /**
  * A combination of onnxSetIO and onnxRunGraph, functionally equals to first run
