Add API version getter to the graph compiler

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

Author: dchigarev

src/dnnl/dnnl_graph_compiler.cpp

@@ -1,9 +1,21 @@
 #include "dnnl_graph_compiler.hpp"
+#include "dnnl_graph_compiler_version.h"
 #include <new>
 
 // dnnl_graph_compiler.h interface implementation.
 // TODO: Implement
 
+dnnl_status_t
+dnnl_graph_compiler_get_api_version(dnnl_graph_compiler_api_version *v) {
+  if (!v)
+    return dnnl_invalid_arguments;
+  v->major = GC_API_VERSION_MAJOR;
+  v->minor = GC_API_VERSION_MINOR;
+  v->patch = GC_API_VERSION_PATCH;
+  v->hash = GC_API_VERSION_HASH;
+  return dnnl_success;
+}
+
 dnnl_status_t
 dnnl_graph_compiler_create(const struct dnnl_graph_compiler_context *ctx,
                            const struct dnnl_graph_compiler **gc) {

test/dnnl/test_dnnl_c_interface.cpp

@@ -1,4 +1,5 @@
 #include "dnnl_graph_compiler.h"
+#include "dnnl_graph_compiler_version.h"
 #include "dnnl_test_utils.hpp"
 #include <gtest/gtest.h>
 
@@ -17,7 +18,7 @@ TEST(dnnl_graph_compiler, c_interface) {
   dnnl_graph_compiler_tensor inputs[2];
   dnnl_graph_compiler_tensor outputs[1];
   uint8_t data_buf[160];
-  size_t dims[1] = {10};
+  int64_t dims[1] = {10};
   inputs[0] = {.id = 0, .ndims = 1, .dims = dims, .data = data_buf};
   inputs[1] = {.id = 1, .ndims = 1, .dims = dims, .data = &data_buf[40]};
   outputs[0] = {.id = 2, .ndims = 1, .dims = dims, .data = &data_buf[80]};
@@ -29,6 +30,17 @@ TEST(dnnl_graph_compiler, c_interface) {
   dnnl_graph_compiler_destroy(gc);
 }
 
+TEST(dnnl_graph_compiler, get_api_version) {
+  ASSERT_EQ(dnnl_graph_compiler_get_api_version(NULL), dnnl_invalid_arguments);
+
+  dnnl_graph_compiler_api_version v;
+  ASSERT_EQ(dnnl_graph_compiler_get_api_version(&v), dnnl_success);
+  ASSERT_EQ(v.major, GC_API_VERSION_MAJOR);
+  ASSERT_EQ(v.minor, GC_API_VERSION_MINOR);
+  ASSERT_EQ(v.patch, GC_API_VERSION_PATCH);
+  ASSERT_STREQ(v.hash, GC_API_VERSION_HASH);
+}
+
 int main(int argc, char **argv) {
   ::testing::InitGoogleTest(&argc, argv);
   return RUN_ALL_TESTS();