The custom kernel requires blas

frost-beta · Nov 15, 2024 · c396ac7 · c396ac7
1 parent 56074ca
commit c396ac7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -174,6 +174,8 @@ if(TORCH_BACKEND_XNNPACK)
 endif()
 if(TORCH_KERNELS_CUSTOM)
   target_link_libraries(${PROJECT_NAME} PRIVATE
+                        "${TORCH_LIBS}/libcpublas.a"
+                        "${TORCH_LIBS}/libeigen_blas.a"
                         "${TORCH_LIBS}/libportable_kernels.a"
                         "${TORCH_LIBS}/libportable_ops_lib.a")
   target_force_link_libraries(${PROJECT_NAME} PRIVATE

diff --git a/README.md b/README.md
@@ -66,7 +66,7 @@ $ wget https://huggingface.co/frost-beta/mobilenet-v2-executorch-cpu/resolve/mai
 Run following code with Node.js:
 
 ```typescript
-import {Module, Tensor} from './dist/index.js';
+import {Module, Tensor} from 'executorch';
 
 // A tensor of shape [ 1, 3, 224, 224 ].
 const input = Array.from({length: 1}, () =>
@@ -134,6 +134,7 @@ export declare enum DType {
     Int8,
     Int16,
     Int32,
+    Int64,
     Float16,
     Float32,
     Float64,