Use parent when fill! a triangular matrix (#278)

* use `parent` when `fill!` a triangular matrix * bump patch
TuringLang · Jul 25, 2023 · 74d52d4 · 74d52d4 · yebai · Jul 25, 2023
1 parent 9cd5907
commit 74d52d4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "Bijectors"
 uuid = "76274a88-744f-5084-9051-94815aaf08c4"
-version = "0.13.2"
+version = "0.13.3"
 
 [deps]
 ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197"

diff --git a/src/utils.jl b/src/utils.jl
@@ -30,7 +30,7 @@ function triu_mask(X::AbstractMatrix, k::Int)
 
  # Using `similar` allows us to respect device of array, etc., e.g. `CuArray`.
  m = similar(X, Bool)
- return triu!(fill!(m, true), k)
+ return triu!(fill!(parent(m), true), k)
 end
 
 ChainRulesCore.@non_differentiable triu_mask(X::AbstractMatrix, k::Int)