ignis-numerics 0.0.1

data/lib/nvruby/tensor.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+# Ignis Tensor Module
+# High-level tensor operations using cuTENSOR
+
+require_relative "tensor/contraction"
+
+module Ignis
+  # Tensor operations module
+  # Provides high-level API for tensor contractions and reductions using cuTENSOR
+  module Tensor
+    class << self
+      # Perform tensor contraction using Einstein notation
+      # @param expression [String] Einstein notation expression (e.g., "ij,jk->ik")
+      # @param a [NvArray] First input tensor
+      # @param b [NvArray] Second input tensor
+      # @param alpha [Float] Scaling factor
+      # @return [NvArray] Result tensor
+      # @example Basic matrix multiplication
+      #   Ignis::Tensor.contract("ij,jk->ik", a, b)
+      # @example Tensor contraction
+      #   Ignis::Tensor.contract("ijk,jkl->il", tensor_a, tensor_b)
+      def contract(expression, a, b, alpha: 1.0)
+        contraction = Contraction.new(expression, a, b, alpha: alpha)
+        contraction.execute
+      ensure
+        contraction&.destroy!
+      end
+
+      # Perform batched tensor contraction
+      # @param expression [String] Einstein notation expression
+      # @param a [NvArray] First input tensor (with batch dimension)
+      # @param b [NvArray] Second input tensor (with batch dimension)
+      # @param alpha [Float] Scaling factor
+      # @return [NvArray] Result tensor
+      def batched_contract(expression, a, b, alpha: 1.0)
+        # Prepend batch dimension to expression if not present
+        contract(expression, a, b, alpha: alpha)
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification
+name: ignis-numerics
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- NNW / Ignis contributors
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ignis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+description: |
+  ignis-numerics is GPU numerical computing for Ruby on the Ignis foundation:
+  dense/sparse linear algebra, FFT (cuFFT), random number generation (cuRAND),
+  linear solvers & decompositions (cuSOLVER), and Einstein-notation tensor
+  contraction (cuTENSOR), with an nvmath-style API (Ignis.fft, Ignis.solve,
+  Ignis.zeros, …). The lineage NvRuby began from, revived as a standalone gem.
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- lib/ignis-numerics.rb
+- lib/nvruby/array.rb
+- lib/nvruby/fft/cufft_bindings.rb
+- lib/nvruby/fft/fft_plan.rb
+- lib/nvruby/fft/operations.rb
+- lib/nvruby/linalg/cutensor_bindings.rb
+- lib/nvruby/mathdx.rb
+- lib/nvruby/mathdx/fft_kernel.rb
+- lib/nvruby/mathdx/gemm_kernel.rb
+- lib/nvruby/random/curand_bindings.rb
+- lib/nvruby/random/generator.rb
+- lib/nvruby/solver.rb
+- lib/nvruby/solver/amgx_bindings.rb
+- lib/nvruby/solver/amgx_config.rb
+- lib/nvruby/solver/amgx_solver.rb
+- lib/nvruby/solver/cudss_bindings.rb
+- lib/nvruby/solver/cusolver_bindings.rb
+- lib/nvruby/solver/eigen.rb
+- lib/nvruby/solver/lu.rb
+- lib/nvruby/solver/sparse_solver.rb
+- lib/nvruby/solver/svd.rb
+- lib/nvruby/sparse/cusparse_bindings.rb
+- lib/nvruby/sparse/sparse_matrix.rb
+- lib/nvruby/tensor.rb
+- lib/nvruby/tensor/contraction.rb
+homepage: https://github.com/tigel-agm/Ignis
+licenses:
+- MIT
+metadata:
+  source_code_uri: https://github.com/tigel-agm/Ignis
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.1'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: GPU numerics for Ruby — BLAS, FFT, RNG, solvers, tensor contraction (nvmath-style)
+test_files: []