[ruby/matrix] Make frozen matrices Ractor shareable

ruby · Dec 9, 2020 · a7dccd0 · a7dccd0
1 parent 6b264e8
commit a7dccd0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/lib/matrix.rb b/lib/matrix.rb
@@ -532,7 +532,8 @@ def collect!(which = :all)
   alias map! collect!
 
   def freeze
-    @rows.freeze
+    @rows.each(&:freeze).freeze
+
     super
   end
 
@@ -2141,6 +2142,9 @@ def zero?
     all?(&:zero?)
   end
 
+  #
+  # Makes the matrix frozen and Ractor-shareable
+  #
   def freeze
     @elements.freeze
     super

diff --git a/test/matrix/test_matrix.rb b/test/matrix/test_matrix.rb
@@ -818,4 +818,13 @@ def test_adjoint
     assert_equal(Matrix[[(1-2i), 1], [(0-1i), 2], [0, 3]], @c1.adjoint)
     assert_equal(Matrix.empty(0,2), @e1.adjoint)
   end
+
+  def test_ractor
+    obj1 = @m1.freeze
+
+    obj2 = Ractor.new obj1 do |obj|
+      obj
+    end.take
+    assert_same obj1, obj2
+  end if defined?(Ractor)
 end