torch-rb 0.13.2 → 0.14.0

data/ext/torch/torch.cpp

@@ -2,6 +2,8 @@
 
 #include <rice/rice.hpp>
 
+#include <fstream>
+
 #include "torch_functions.h"
 #include "templates.h"
 #include "utils.h"
@@ -57,16 +59,18 @@ void init_torch(Rice::Module& m) {
       "_save",
       [](const torch::IValue &value) {
         auto v = torch::pickle_save(value);
-        std::string str(v.begin(), v.end());
-        return str;
+        return Object(rb_str_new(v.data(), v.size()));
       })
     .define_singleton_function(
       "_load",
-      [](const std::string &s) {
-        std::vector<char> v;
-        std::copy(s.begin(), s.end(), std::back_inserter(v));
+      [](const std::string &filename) {
         // https://github.com/pytorch/pytorch/issues/20356#issuecomment-567663701
-        return torch::pickle_load(v);
+        std::ifstream input(filename, std::ios::binary);
+        std::vector<char> bytes(
+            (std::istreambuf_iterator<char>(input)),
+            (std::istreambuf_iterator<char>()));
+        input.close();
+        return torch::pickle_load(bytes);
       })
     .define_singleton_function(
       "_from_blob",

data/ext/torch/utils.h

@@ -6,7 +6,7 @@
 #include <rice/stl.hpp>
 
 static_assert(
-  TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR == 0,
+  TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR == 1,
   "Incompatible LibTorch version"
 );
 

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.13.2"
+  VERSION = "0.14.0"
 end

data/lib/torch.rb

@@ -387,8 +387,11 @@ module Torch
       File.binwrite(f, _save(to_ivalue(obj)))
     end
 
-    def load(f)
-      to_ruby(_load(File.binread(f)))
+    def load(filename)
+      # keep backwards compatibility
+      File.open(filename, "rb") { |f| f.read(1) }
+
+      to_ruby(_load(filename))
     end
 
     def tensor(data, **options)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.13.2
+  version: 0.14.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-12 00:00:00.000000000 Z
+date: 2023-11-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice