ignis-numerics 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: cb5424db880779427450a67cf0467f1fdccff7cdf1e60bf9e51d25f946502983
+  data.tar.gz: 61d09d1019ce4d4113901bc2abf5605260dcf34f5ceda5b7481cd3eb57a5f928
+SHA512:
+  metadata.gz: f58e78ef1d9b2d401585717a301b184fba8252143060330e1f9bb33eb7f1035c959fbe1ec044c89551a153e4b7a1f2c8fe4573ed2b4019f1f499fbfb5be63c85
+  data.tar.gz: 9b9565f1ef0404fc2dcd7feb5c31c77b36b9e81827daf28e1f91badf23eb1e00ff5ab8190802c862465f63eb90c54f00d7baae8d18b08b6f1904fc8862461403

data/README.md

@@ -0,0 +1,15 @@
+# ignis-numerics
+
+GPU numerical computing for Ruby (nvmath-style), on the [`ignis`](https://rubygems.org/gems/ignis) foundation. The lineage this project began from — an nvmath-python port — revived as a standalone gem.
+
+Dense/sparse linear algebra, FFT (cuFFT), RNG (cuRAND), solvers & decompositions (cuSOLVER), and Einstein-notation tensor contraction (cuTENSOR).
+
+```ruby
+require "ignis-numerics"
+a = Ignis::NvArray.from_array([[1.0, 2.0], [3.0, 4.0]], device: 0)
+Ignis.matmul(a, a).to_a          # GPU GEMM
+Ignis.fft(signal)                # cuFFT
+Ignis.solve(matrix, rhs)         # cuSOLVER
+```
+
+Requires an NVIDIA GPU + CUDA runtime (cuFFT/cuRAND/cuSOLVER/cuSPARSE/cuTENSOR). MIT.

data/lib/ignis-numerics.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+# ignis-numerics — nvmath-for-Ruby: GPU dense/sparse linear algebra, FFT, RNG, and
+# tensor contraction (cuBLAS/cuFFT/cuRAND/cuSOLVER/cuSPARSE/cuTENSOR), built on the
+# Ignis GPU foundation. This is the lineage NvRuby started from (an nvmath-python
+# port), revived as a standalone gem. Provides the numerics array Ignis::NvArray and
+# an nvmath-style module API (Ignis.fft, Ignis.solve, Ignis.zeros, …).
+
+require "ignis"
+
+# Numerics GPU array + creation helpers
+require_relative "nvruby/array"
+
+# cuFFT
+require_relative "nvruby/fft/cufft_bindings"
+require_relative "nvruby/fft/operations"
+require_relative "nvruby/fft/fft_plan"
+
+# cuRAND
+require_relative "nvruby/random/curand_bindings"
+require_relative "nvruby/random/generator"
+
+# cuSPARSE
+require_relative "nvruby/sparse/cusparse_bindings"
+require_relative "nvruby/sparse/sparse_matrix"
+
+# cuSOLVER
+require_relative "nvruby/solver"
+
+# cuTENSOR
+require_relative "nvruby/linalg/cutensor_bindings"
+require_relative "nvruby/tensor"
+
+# MathDx (device extensions)
+require_relative "nvruby/mathdx"
+
+module Ignis
+  class << self
+    # --- Array creation (nvmath-style) ---
+    def zeros(shape, dtype: :float32, device: nil); NvArray.zeros(shape, dtype: dtype, device: device); end
+    def ones(shape, dtype: :float32, device: nil);  NvArray.ones(shape, dtype: dtype, device: device);  end
+    def eye(size, dtype: :float32, device: nil);     NvArray.eye(size, dtype: dtype, device: device);    end
+    def array(data, dtype: :float32, device: nil);   NvArray.from_array(data, dtype: dtype, device: device); end
+    def linspace(a, b, num, dtype: :float32, device: nil); NvArray.linspace(a, b, num, dtype: dtype, device: device); end
+
+    # --- FFT (cuFFT) ---
+    def fft(x, axis: -1, norm: :backward);   FFT::Operations.fft(x, axis: axis, norm: norm);   end
+    def ifft(x, axis: -1, norm: :backward);  FFT::Operations.ifft(x, axis: axis, norm: norm);  end
+    def rfft(x, n: nil, axis: -1, norm: :backward);  FFT::Operations.rfft(x, n: n, axis: axis, norm: norm); end
+    def irfft(x, n: nil, axis: -1, norm: :backward); FFT::Operations.irfft(x, n: n, axis: axis, norm: norm); end
+
+    # --- Linear solvers / decompositions (cuSOLVER) ---
+    def solve(a, b);        Solver.solve(a, b);   end
+    def lu(m);              Solver.lu(m);          end
+    def svd(m, full_matrices: false); Solver.svd(m, full_matrices: full_matrices); end
+    def eigh(m, eigenvectors: true);  Solver.eigh(m, eigenvectors: eigenvectors);  end
+    def eig(m);             Solver.eig(m);         end
+
+    # --- Tensor contraction (cuTENSOR) ---
+    def contract(expr, a, b, alpha: 1.0); Tensor.contract(expr, a, b, alpha: alpha); end
+  end
+end