From ac68eb0f75e8d61f41a1fedf44be4d1d9aa70c29 Mon Sep 17 00:00:00 2001 From: Penelope Yong Date: Mon, 16 Dec 2024 16:39:09 +0000 Subject: [PATCH] Update Mooncake (#568) --- Manifest.toml | 8 ++++---- tutorials/03-bayesian-neural-network/index.qmd | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/Manifest.toml b/Manifest.toml index 6c38addbd..980a27805 100644 --- a/Manifest.toml +++ b/Manifest.toml @@ -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" @@ -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" diff --git a/tutorials/03-bayesian-neural-network/index.qmd b/tutorials/03-bayesian-neural-network/index.qmd index 7e21fce37..1b2033224 100755 --- a/tutorials/03-bayesian-neural-network/index.qmd +++ b/tutorials/03-bayesian-neural-network/index.qmd @@ -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