ext/torch/extconf.rb in torch-rb-0.1.4 vs ext/torch/extconf.rb in torch-rb-0.1.5
- old
+ new
@@ -20,6 +20,12 @@
$LDFLAGS << " -Wl,-rpath,#{lib}"
$LDFLAGS << " -L#{lib}"
$LDFLAGS << " -ltorch -lc10"
+# generate C++ functions
+puts "Generating C++ functions..."
+require_relative "../../lib/torch/native/generator"
+Torch::Native::Generator.generate_cpp_functions
+
+# create makefile
create_makefile("torch/ext")