Updated LibTorch to 2.1.0

ankane · Nov 9, 2023 · daf3fce · daf3fce
1 parent cf6cba3
commit daf3fce
Show file tree

Hide file tree

Showing 6 changed files with 998 additions and 521 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -14,9 +14,9 @@ jobs:
           - ruby: "3.0"
             os: ubuntu-20.04
     env:
-      LIBTORCH_VERSION: 2.0.1
+      LIBTORCH_VERSION: 2.1.0
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - uses: ruby/setup-ruby@v1
         with:
           ruby-version: ${{ matrix.ruby }}

diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,6 @@
-## 0.13.3 (unreleased)
+## 0.14.0 (unreleased)
 
+- Updated LibTorch to 2.1.0
 - Improved performance of saving and loading models
 
 ## 0.13.2 (2023-05-11)

diff --git a/codegen/generate_functions.rb b/codegen/generate_functions.rb
@@ -48,7 +48,12 @@ def skip_functions(functions)
     f.base_name == "split_with_sizes_copy" ||
     f.base_name == "unbind_copy" ||
     # TODO fix LibTorch 1.13 changes
-    f.base_name == "native_channel_shuffle"
+    f.base_name == "native_channel_shuffle" ||
+    # TODO fix LibTorch 2.1 changes
+    f.base_name == "sym_size" ||
+    f.base_name == "sym_numel" ||
+    f.base_name == "sym_storage_offset" ||
+    f.base_name == "sym_stride"
   end
 end