Sha256: d4b7b3b666afa7e67d99acaca88d56e5b45d4de6b1518b016d72ab3c0d9132b4
Contents?: true
Size: 806 Bytes
Versions: 1
Compression:
Stored size: 806 Bytes
Contents
require "mkmf-rice" abort "Missing stdc++" unless have_library("stdc++") $CXXFLAGS << " -std=c++11" # needed for Linux pre-cxx11 ABI version # $CXXFLAGS << " -D_GLIBCXX_USE_CXX11_ABI=0" # silence ruby/intern.h warning $CXXFLAGS << " -Wno-deprecated-register" # silence torch warnings $CXXFLAGS << " -Wno-shorten-64-to-32 -Wno-missing-noreturn" inc, lib = dir_config("torch") inc ||= "/usr/local/include" lib ||= "/usr/local/lib" $INCFLAGS << " -I#{inc}" $INCFLAGS << " -I#{inc}/torch/csrc/api/include" $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")
Version data entries
1 entries across 1 versions & 1 rubygems
Version | Path |
---|---|
torch-rb-0.1.8 | ext/torch/extconf.rb |