torch-rb 0.1.8 → 0.2.0

data/ext/torch/torch_functions.hpp

@@ -0,0 +1,6 @@
+// generated by rake generate:functions
+// do not edit by hand
+
+#pragma once
+
+void add_torch_functions(Module m);

data/lib/torch/hub.rb

@@ -0,0 +1,9 @@
+module Torch
+  module Hub
+    class << self
+      def list(github, force_reload: false)
+        raise NotImplementedYet
+      end
+    end
+  end
+end

data/lib/torch/native/generator.rb

@@ -18,14 +18,12 @@ module Torch
           functions = functions()
 
           # skip functions
-          skip_binding = ["unique_dim_consecutive", "einsum", "normal"]
           skip_args = ["bool[3]", "Dimname", "MemoryFormat", "Layout", "Storage", "ConstQuantizerPtr"]
 
           # remove functions
           functions.reject! do |f|
             f.ruby_name.start_with?("_") ||
             f.ruby_name.end_with?("_backward") ||
-            skip_binding.include?(f.ruby_name) ||
             f.args.any? { |a| a[:type].include?("Dimname") }
           end
 
@@ -34,7 +32,10 @@ module Torch
             functions.partition do |f|
               f.args.any? do |a|
                 a[:type].include?("?") && !["Tensor?", "Generator?", "int?", "ScalarType?"].include?(a[:type]) ||
-                skip_args.any? { |sa| a[:type].include?(sa) }
+                skip_args.any? { |sa| a[:type].include?(sa) } ||
+                # native_functions.yaml is missing size argument for normal
+                # https://pytorch.org/cppdocs/api/function_namespacetorch_1a80253fe5a3ded4716ec929a348adb4b9.html
+                (f.base_name == "normal" && !f.out?)
               end
             end
 
@@ -119,6 +120,8 @@ void add_%{type}_functions(Module m) {
                   "IntArrayRef"
                 when /Tensor\(\S!?\)/
                   "Tensor &"
+                when "str"
+                  "std::string"
                 else
                   a[:type]
                 end