tensor 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ab1431a30c46e725f4719be6d7af3f06f77720741f858448528f23528452baea
-  data.tar.gz: 2282380204a6d8b8a839365b265cb48443dd16e1e6206bff1d6d7c68af573421
+  metadata.gz: 3f884993f373ad40adb3118ac97556fc308a7550e2a54e454ecd60f3dc039f75
+  data.tar.gz: f76816b84f9db6b0dc91b4ea7e23671aacb3bb2347e0942b857fbfd2bf726b2e
 SHA512:
-  metadata.gz: 785a6cf18705e00c2bad28b89a1b2682b139f2423f3ba683fdfef5f2bf3204b4f1a5681c47d2734f8235585bc36e20d2c80790e0abfbf0ee5f625e2384728361
-  data.tar.gz: 7cdea95dba2a3169e10d8b1d425ada3bb06113dfca07d63287d7cc54f9014a025d31adb8a445375c5f9b00c96d5ae0532d39c39d588211f5308a36caa9743596
+  metadata.gz: 87bb73be74447c67802aba582f358f8489eaac06d7c0092efc660e977c6781f8901931a0e5522869bd03434af56340851eeeef2bfb094918bad3fbe4c790d2b3
+  data.tar.gz: 1176c4623cc43bc70bb04e06132cb143460d7e5470ab14bbc5ba564535b464a931cc9a77707008346ea832c9b17348f9ac464ef91768e884667cda41210548b0

data/extconf.rb

@@ -6,11 +6,23 @@ $VERBOSE = true
 
 host_os   = RbConfig::CONFIG['host_os']
 on_macos  = host_os =~ /darwin/
+on_linux  = host_os =~ /linux/
+
+# Brief dependency summary for users on all platforms
+puts "\n== tensor native extension build =="
+puts "Host OS: #{host_os}"
+puts "This extension can use (optionally):"
+puts "  - OpenMP (libomp/libgomp) for multi-threading"
+puts "  - OpenBLAS (openblas) for BLAS optimizations"
+puts "Environment overrides:"
+puts "  - OPENBLAS_DIR  : root with include/ and lib/ for OpenBLAS"
+puts "  - OMP_INCLUDE_DIR / OMP_LIB_DIR: OpenMP headers/libs"
+puts "  - PKG_CONFIG     : path to pkg-config\n"
 
 # Debug environment variables
 puts "PKG_CONFIG_PATH: #{ENV['PKG_CONFIG_PATH']}"
-puts "PATH: #{ENV['PATH']}"
-puts "pkg-config location: #{`which pkg-config`.chomp}"
+puts "PATH:           #{ENV['PATH']}"
+puts "pkg-config:     #{`which pkg-config`.chomp}"
 
 # Set absolute path to pkg-config if needed (primarily for Homebrew on macOS)
 if on_macos && ENV['PKG_CONFIG'].to_s.empty?
@@ -31,30 +43,49 @@ is_x86 = RbConfig::CONFIG['host_cpu'] =~ /x86|x64|i386/
 # Detect if we're using Apple's clang
 is_apple_clang = RbConfig::CONFIG['CC'] =~ /^clang/ && RUBY_PLATFORM =~ /darwin/
 
-# OpenMP support for macOS
+# OpenMP support (optional)
+omp_include_dir = ENV['OMP_INCLUDE_DIR']
+omp_lib_dir     = ENV['OMP_LIB_DIR']
+
+if omp_include_dir && !omp_include_dir.empty?
+  $INCFLAGS << " -I#{omp_include_dir}"
+end
+if omp_lib_dir && !omp_lib_dir.empty?
+  $LDFLAGS << " -L#{omp_lib_dir}"
+end
+
+openmp_enabled = false
+
 if is_apple_clang
-  puts "Detected Apple's clang, using LLVM OpenMP..."
-  # Use LLVM's OpenMP
+  puts "Detected Apple's clang, trying LLVM OpenMP..."
   libomp_prefix = "/opt/homebrew/opt/libomp"
   if Dir.exist?(libomp_prefix)
     $INCFLAGS << " -I#{libomp_prefix}/include"
     $LDFLAGS << " -L#{libomp_prefix}/lib"
     $CFLAGS << " -Xpreprocessor -fopenmp"
     $LDFLAGS << " -lomp"
-    have_library('omp')
-    puts "LLVM OpenMP support enabled"
+    openmp_enabled = have_library('omp')
+    puts "LLVM OpenMP support #{openmp_enabled ? 'enabled' : 'not found'}"
+    unless openmp_enabled
+      puts "To enable: brew install libomp"
+    end
   else
-    puts "Warning: OpenMP not found. Some optimizations will be disabled."
-    puts "To enable OpenMP: brew install libomp"
+    puts "LLVM libomp not found under /opt/homebrew/opt/libomp."
+    puts "To enable OpenMP on macOS: brew install libomp"
   end
 else
   # Standard OpenMP support for non-Apple compilers
   if have_library('omp') || have_library('gomp')
     $CFLAGS << " -fopenmp"
     $LDFLAGS << " -fopenmp"
-    puts "OpenMP support enabled"
+    openmp_enabled = true
+    puts "OpenMP support enabled via -fopenmp"
   else
-    puts "Warning: OpenMP not found. Some optimizations will be disabled."
+    puts "OpenMP library (libomp/libgomp) not found."
+    if on_linux
+      puts "On Debian/Ubuntu: sudo apt-get install libomp-dev (or ensure gcc's libgomp is available)."
+      puts "On Fedora/RHEL:   sudo dnf install libgomp"
+    end
   end
 end
 
@@ -62,29 +93,29 @@ end
 openblas_include_path = nil
 openblas_lib_path     = nil
 
-if on_macos
-  # Allow overriding OpenBLAS location, otherwise try Homebrew cellars
-  explicit_openblas = ENV['OPENBLAS_DIR']
-  if explicit_openblas && Dir.exist?(explicit_openblas)
-    openblas_root = explicit_openblas
-  else
-    openblas_versions = Dir.glob("/opt/homebrew/Cellar/openblas/*").sort
-    openblas_root = openblas_versions.last
-  end
+explicit_openblas = ENV['OPENBLAS_DIR']
+if explicit_openblas && Dir.exist?(explicit_openblas)
+  openblas_root = explicit_openblas
+elsif on_macos
+  openblas_versions = Dir.glob("/opt/homebrew/Cellar/openblas/*").sort
+  openblas_root = openblas_versions.last
+else
+  openblas_root = nil
+end
 
-  if openblas_root && Dir.exist?(openblas_root)
-    openblas_include_path = File.join(openblas_root, "include")
-    openblas_lib_path     = File.join(openblas_root, "lib")
-    $INCFLAGS << " -I#{openblas_include_path}"
-    $LDFLAGS << " -L#{openblas_lib_path}"
-    $LIBPATH << openblas_lib_path
-  else
-    puts "OpenBLAS not found under /opt/homebrew/Cellar; BLAS optimizations may be disabled."
-  end
+if openblas_root && Dir.exist?(openblas_root)
+  openblas_include_path = File.join(openblas_root, "include")
+  openblas_lib_path     = File.join(openblas_root, "lib")
+  $INCFLAGS << " -I#{openblas_include_path}"
+  $LDFLAGS << " -L#{openblas_lib_path}"
+  $LIBPATH << openblas_lib_path
+  puts "Using OpenBLAS from #{openblas_root}"
+elsif explicit_openblas
+  puts "OPENBLAS_DIR=#{explicit_openblas} was set but does not exist; ignoring."
 end
 
 # Check for BLAS support using pkg-config (cross-platform)
-puts "Checking for BLAS support..."
+puts "Checking for BLAS (OpenBLAS) support..."
 if pkg_config("openblas")
   puts "BLAS support found via pkg-config! Enabling BLAS optimizations."
 elsif openblas_lib_path && Dir.exist?(openblas_lib_path)
@@ -92,6 +123,11 @@ elsif openblas_lib_path && Dir.exist?(openblas_lib_path)
   $LDFLAGS << " -lopenblas"
 else
   puts "OpenBLAS not found. Continuing without BLAS-specific optimizations."
+  if on_linux
+    puts "On Debian/Ubuntu: sudo apt-get install libopenblas-dev"
+    puts "On Fedora/RHEL:   sudo dnf install openblas-devel"
+  end
+  puts "Or set OPENBLAS_DIR to your OpenBLAS installation root (with include/ and lib/)."
 end
 
 # Clean up incompatible OpenMP flags from pkg-config when using Apple's clang
@@ -127,9 +163,11 @@ $CFLAGS << " -fno-common -pipe"
 $CFLAGS << " -D_XOPEN_SOURCE -D_DARWIN_C_SOURCE -D_DARWIN_UNLIMITED_SELECT -D_REENTRANT"
 
 # Additional include and library paths
-$INCFLAGS << " -I/opt/homebrew/include"
+if on_macos
+  $INCFLAGS << " -I/opt/homebrew/include"
+  $LDFLAGS << " -L/opt/homebrew/lib"
+end
 $INCFLAGS << " -I/usr/local/include"
-$LDFLAGS << " -L/opt/homebrew/lib"
 $LDFLAGS << " -L/usr/local/lib"
 
 # Ensure proper linking for neural network operations

metadata

@@ -1,10 +1,10 @@
 --- !ruby/object:Gem::Specification
 name: tensor
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
-- Your Name
+- Pankaj Doharey
 bindir: bin
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
@@ -12,7 +12,7 @@ dependencies: []
 description: A native Ruby C extension providing parallelized matrix and neural-network
   style tensor operations.
 email:
-- you@example.com
+- pankajdoharey@gmail.com
 executables: []
 extensions:
 - extconf.rb
@@ -40,7 +40,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 4.0.1
 specification_version: 4
 summary: Fast tensor/matrix operations for Ruby using OpenMP and OpenBLAS
 test_files: []