Skip to content

Commit

Permalink
Update Mooncake (#568)
Browse files Browse the repository at this point in the history
  • Loading branch information
penelopeysm authored Dec 16, 2024
1 parent 5d98ede commit ac68eb0
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions Manifest.toml
Original file line number Diff line number Diff line change
Expand Up @@ -654,9 +654,9 @@ version = "0.4.0"

[[deps.DiffEqBase]]
deps = ["ArrayInterface", "ConcreteStructs", "DataStructures", "DocStringExtensions", "EnumX", "EnzymeCore", "FastBroadcast", "FastClosures", "FastPower", "ForwardDiff", "FunctionWrappers", "FunctionWrappersWrappers", "LinearAlgebra", "Logging", "Markdown", "MuladdMacro", "Parameters", "PreallocationTools", "PrecompileTools", "Printf", "RecursiveArrayTools", "Reexport", "SciMLBase", "SciMLOperators", "SciMLStructures", "Setfield", "Static", "StaticArraysCore", "Statistics", "TruncatedStacktraces"]
git-tree-sha1 = "b7dbeaa770bad0980ddddf606de814cff2acb3bc"
git-tree-sha1 = "b1e23a7fe7371934d9d538114a7e7166c1d09e05"
uuid = "2b5f629d-d688-5b77-993f-72d75c75574e"
version = "6.160.0"
version = "6.161.0"

[deps.DiffEqBase.extensions]
DiffEqBaseCUDAExt = "CUDA"
Expand Down Expand Up @@ -2159,9 +2159,9 @@ version = "0.8.1"

[[deps.Mooncake]]
deps = ["ADTypes", "ChainRules", "ChainRulesCore", "DiffRules", "DiffTests", "ExprTools", "FunctionWrappers", "Graphs", "InteractiveUtils", "LinearAlgebra", "MistyClosures", "Random", "Setfield", "Test"]
git-tree-sha1 = "46c0f7c9cbe46232d0f4f2145126bf7b3499ffd0"
git-tree-sha1 = "86238d75d47d0cae923504e201c8fc00b797705f"
uuid = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
version = "0.4.62"
version = "0.4.63"

[deps.Mooncake.extensions]
MooncakeAllocCheckExt = "AllocCheck"
Expand Down
2 changes: 1 addition & 1 deletion tutorials/03-bayesian-neural-network/index.qmd
Original file line number Diff line number Diff line change
Expand Up @@ -169,7 +169,7 @@ function vector_to_parameters(ps_new::AbstractVector, ps::NamedTuple)
@assert length(ps_new) == Lux.parameterlength(ps)
i = 1
function get_ps(x)
z = reshape(ps_new[i:(i + length(x) - 1)], size(x))
z = reshape(view(ps_new, i:(i + length(x) - 1)), size(x))
i += length(x)
return z
end
Expand Down

0 comments on commit ac68eb0

Please sign in to comment.