Feature/cpu

.gitignore

@@ -68,7 +68,7 @@ artifacts/
 *.pidb
 *.svclog
 *.scc
-
+*.bin
 # Chutzpah Test files
 _Chutzpah*
 
@@ -306,4 +306,4 @@ cmake-build-*
 *gmodel_dump_dir*
 *.ipynb_checkpoints*
 # Auto generated files
-# generated/
+# generated/

CMakeLists.txt

@@ -12,7 +12,6 @@ endif()
 
 if(NOT DEFINED NNCASE_VERSION_SUFFIX)
     find_package (Git)
-
     execute_process(
         COMMAND ${GIT_EXECUTABLE} describe --always --dirty --tag
         WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
@@ -274,5 +273,5 @@ if(BUILD_TESTING)
 endif()
 
 # Modules
-#add_subdirectory(modules/k210)
+
 #add_subdirectory(modules/vulkan)

Directory.Packages.props

@@ -49,8 +49,9 @@
     <PackageVersion Include="OrtKISharp" Version="0.0.2" />
     <PackageVersion Include="RazorLight" Version="2.3.0" />
     <PackageVersion Include="Singulink.Collections.Weak" Version="1.0.2" />
-    <PackageVersion Include="StyleCop.Analyzers" Version="1.2.0-beta.507" />
-    <PackageVersion Include="System.CommandLine.Hosting" Version="0.3.0-alpha.21216.1" />
+    <PackageVersion Include="StyleCop.Analyzers" Version="1.2.0-beta.435" />
+    <PackageVersion Include="System.CommandLine.Hosting" Version="0.4.0-alpha.22272.1" />
+    <PackageVersion Include="System.CommandLine" Version="2.0.0-beta4.22272.1" />
     <PackageVersion Include="System.Linq.Async" Version="6.0.1" />
     <PackageVersion Include="System.Reactive" Version="5.0.0" />
     <PackageVersion Include="Tomlyn.Extensions.Configuration" Version="1.0.5" />

modules/Nncase.Modules.StackVM/CodeGen/StackVM/CodeGenVisitor.g.cs

@@ -1,6 +1,6 @@
 // Copyright (c) Canaan Inc. All rights reserved.
 // Licensed under the Apache license. See LICENSE file in the project root for full license information.
-/* This file is generated by tools/stackvm_gen/IsaGen at 2023/9/18 下午5:04:31 +08:00. */
+/* This file is generated by tools/stackvm_gen/IsaGen at 9/20/2023 10:17:08 AM +00:00. */
 
 using System;
 using System.Collections.Generic;
@@ -59,7 +59,7 @@
                 Emitter.T.L2Normalization();
                 break;
             case IR.NN.LayerNorm top:
-                Emitter.T.LayerNorm(top.Axis, top.Epsilon);
+                Emitter.T.LayerNorm(top.Axis, top.Epsilon, top.UseMean);
                 break;
             case IR.NN.LeakyRelu top:
                 Emitter.T.LeakyRelu();
@@ -176,7 +176,7 @@
                 Emitter.T.Cast(top.NewType, top.CastMode);
                 break;
             case IR.Tensors.Concat top:
-                Emitter.T.Concat();
+                Emitter.T.Concat(top.Axis);
                 break;
             case IR.Tensors.ConstantOfShape top:
                 Emitter.T.ConstantOfShape();
@@ -191,7 +191,7 @@
                 Emitter.T.Flatten();
                 break;
             case IR.Tensors.Gather top:
-                Emitter.T.Gather();
+                Emitter.T.Gather(top.Axis);
                 break;
             case IR.Tensors.GatherElements top:
                 Emitter.T.GatherElements();
@@ -205,9 +205,6 @@
             case IR.Tensors.IndexOf top:
                 Emitter.T.IndexOf();
                 break;
-            case IR.Tensors.LSTM top:
-                Emitter.T.LSTM(top.Direction, top.Layout, top.Activations);
-                break;
             case IR.Tensors.Prod top:
                 Emitter.T.Prod();
                 break;
@@ -289,6 +286,9 @@
             case IR.ShapeExpr.UnsqueezeShape top:
                 Emitter.T.UnsqueezeShape();
                 break;
+            case IR.RNN.LSTM top:
+                Emitter.T.LSTM(top.Direction, top.Layout, top.Activations);
+                break;
             case IR.Random.Normal top:
                 Emitter.T.Normal(top.Type);
                 break;