Merge branch 'main' into im-dev

Author: Iainmon

CMakeLists.txt

@@ -29,7 +29,7 @@ if(APPLE)
     set(CMAKE_C_COMPILER "/usr/bin/clang")
     set(CMAKE_CXX_COMPILER "/usr/bin/clang++")
 endif()
-set(CMAKE_CXX_STANDARD 23)
+set(CMAKE_CXX_STANDARD 20)
 
 
 include(LibTorchDL)
@@ -137,7 +137,7 @@ set(BRIDGE_OBJECT_FILES $<TARGET_OBJECTS:bridge>)
 
 
 file(GLOB LIBTORCH_ALL_LIB_FILES 
-    "${LIBTORCH_DIR}/lib/*.a" 
+    # "${LIBTORCH_DIR}/lib/*.a" 
     "${LIBTORCH_DIR}/lib/*.dylib" 
     "${LIBTORCH_DIR}/lib/*.so")
 
@@ -152,7 +152,6 @@ set(REQUIRED_LIBS
     "libtorch"
     "libtorch_cpu"
     "libc10"
-    "libtorch_global_deps"
 )
 
 set(DISALLOWED_LIBS

bridge/include/bridge.h

@@ -7,6 +7,9 @@
 extern "C" {
 #endif
 
+#define proto_bridge_simple(Name) \
+    bridge_tensor_t Name(bridge_tensor_t input)
+
 typedef float float32_t;
 typedef double float64_t;
 typedef char bool_t;
@@ -77,6 +80,28 @@ bridge_tensor_t max_pool2d(
     int dilation
 );
 
+proto_bridge_simple(relu);
+
+proto_bridge_simple(relu6);
+
+proto_bridge_simple(gelu);
+
+proto_bridge_simple(logsigmoid);
+
+proto_bridge_simple(mish);
+
+proto_bridge_simple(selu);
+
+proto_bridge_simple(silu);
+
+proto_bridge_simple(softmax);
+
+proto_bridge_simple(softmin);
+
+proto_bridge_simple(softsign);
+
+proto_bridge_simple(tanhshrink);
+
 
 // bridge_tensor_t conv2d(
 //     bridge_tensor_t input,
@@ -93,4 +118,4 @@ bridge_tensor_t max_pool2d(
 #endif
 
 #endif // BRIDGE_H
-//hello
+//hello

bridge/lib/bridge.cpp

@@ -14,7 +14,12 @@
 #include <vector>
 #include <cstdint>
 
-
+#define def_bridge_simple(Name) \
+    extern "C" bridge_tensor_t Name(bridge_tensor_t input) { \
+        auto t_input = bridge_to_torch(input); \
+        auto t_output = torch::Name(t_input); \
+        return torch_to_bridge(t_output); \
+    }
 
 
 
@@ -56,18 +61,6 @@ torch::Tensor bridge_to_torch(bridge_tensor_t &bt) {
     return torch::from_blob(bt.data, shape, torch::kFloat);
 }
 
-
-
-
-
-
-
-
-
-
-
-
-
 extern "C" float32_t* unsafe(const float32_t* arr) {
     return const_cast<float32_t*>(arr);
 }
@@ -133,15 +126,6 @@ extern "C" bridge_tensor_t load_run_model(const uint8_t* model_path, bridge_tens
     return torch_to_bridge(output);
 }
 
-
-
-
-
-
-
-
-
-
 extern "C" bridge_tensor_t increment3(bridge_tensor_t arr) {
     auto t = bridge_to_torch(arr);
     // Increment the tensor
@@ -386,4 +370,4 @@ extern "C" float sumArray(float* arr, int* sizes, int dim) {
 
     // auto t = torch::from_blob(arr, shape, torch::kFloat);
     // return t.sum().item<float>();
-}
+}

cmake/LibTorchDL.cmake

@@ -111,4 +111,4 @@ endfunction()
 
 # dlcache("${TORCH_DISTRIBUTION}" OUT url)
 
-# message(STATUS "file loc >>>>> ${url}")
+# message(STATUS "file loc >>>>> ${url}")