Sha256: 5e83e93cd85d36eac739bef60d690b0404819a61a642a748c27cb8b656ab121d

Contents?: true

Size: 720 Bytes

Versions: 3

Compression:

Stored size: 720 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"

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

3 entries across 3 versions & 1 rubygems

Version Path
torch-rb-0.1.7 ext/torch/extconf.rb
torch-rb-0.1.6 ext/torch/extconf.rb
torch-rb-0.1.5 ext/torch/extconf.rb