Update Reactant_jll and bump version number (#984)

* Update Reactant_jll and bump version number

* fix: apply transpose in sharding_constraint

---------

Co-authored-by: Avik Pal <[email protected]>

Author: giordano

Project.toml

@@ -1,7 +1,7 @@
 name = "Reactant"
 uuid = "3c362404-f566-11ee-1572-e11a4b42c853"
 authors = ["William Moses <[email protected]>", "Valentin Churavy <[email protected]>", "Sergio Sánchez Ramírez <[email protected]>", "Paul Berg <[email protected]>", "Avik Pal <[email protected]>", "Mosè Giordano <[email protected]>"]
-version = "0.2.46"
+version = "0.2.47"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -86,7 +86,7 @@ PythonCall = "0.9"
 Random = "1.10"
 Random123 = "1.7"
 ReactantCore = "0.1.6"
-Reactant_jll = "0.0.94"
+Reactant_jll = "0.0.95"
 Scratch = "1.2"
 Sockets = "1.10"
 SpecialFunctions = "2.4"

src/Ops.jl

@@ -2388,7 +2388,7 @@ Produces a [`Reactant.MLIR.Dialects.sdy.sharding_constraint`](@ref) operation wi
     haskey(cache, sharding.mesh) || Ops.mesh(sharding.mesh; location)
     (; sym_name, mesh_attr) = cache[sharding.mesh]
     tensor_sharding_attr = Reactant.Sharding.get_shardy_tensor_sharding_attribute(
-        sharding, MLIR.IR.context(), sym_name, mesh_attr; do_transpose=true
+        sharding, MLIR.IR.context(), sym_name, mesh_attr; do_transpose=false
     )
     resharded_value = MLIR.IR.result(
         MLIR.Dialects.sdy.sharding_constraint(

test/sharding.jl

@@ -209,12 +209,13 @@ end
         @test contains(repr(hlo), "sharding_constraint")
         hlo = @code_hlo shardy_passes = :to_mhlo_shardings fn_with_constraint(x_ra)
         @test !contains(repr(hlo), "sharding_constraint")
-        @test length(collect(eachmatch(r"mhlo.sharding", repr(hlo)))) == 3
+        @test length(collect(eachmatch(r"mhlo.sharding", repr(hlo)))) == 5
 
         z = Reactant.to_rarray(x; sharding=constraint)
         res = @jit fn_with_constraint(x_ra)
 
         @test x .+ x ≈ Array(res)
+
         @test string(z.sharding.sharding.hlo_sharding) ==
             string(res.sharding.sharding.hlo_sharding)
         @test string(res.sharding.sharding.hlo_sharding) !=
@@ -229,7 +230,7 @@ end
             x_ra_no_sharding
         )
         @test !contains(repr(hlo), "sharding_constraint")
-        @test length(collect(eachmatch(r"mhlo.sharding", repr(hlo)))) == 3
+        @test length(collect(eachmatch(r"mhlo.sharding", repr(hlo)))) == 5
 
         res = @jit fn_with_constraint(x_ra_no_sharding)
         @test x .+ x ≈ Array(res)