Using make_mut instead of get_mut

coreylowman · Jul 24, 2023 · 6912801 · 6912801
1 parent 2190ba8
commit 6912801
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/tensor_ops/adam/mod.rs b/src/tensor_ops/adam/mod.rs
@@ -72,7 +72,7 @@ impl AdamConfig {
         param.device.adam_kernel(
             t,
             self,
-            std::sync::Arc::get_mut(&mut param.data).unwrap(),
+            std::sync::Arc::make_mut(&mut param.data),
             moment1,
             moment2,
             grad,

diff --git a/src/tensor_ops/rmsprop/mod.rs b/src/tensor_ops/rmsprop/mod.rs
@@ -69,7 +69,7 @@ impl RMSpropConfig {
     ) -> Result<(), D::Err> {
         param.device.rmsprop_kernel(
             self,
-            std::sync::Arc::get_mut(&mut param.data).unwrap(),
+            std::sync::Arc::make_mut(&mut param.data),
             momentum,
             square_avg,
             grad_avg,

diff --git a/src/tensor_ops/sgd/mod.rs b/src/tensor_ops/sgd/mod.rs
@@ -102,7 +102,7 @@ impl SgdConfig {
     ) -> Result<(), D::Err> {
         param.device.sgd_kernel(
             self,
-            std::sync::Arc::get_mut(&mut param.data).unwrap(),
+            std::sync::Arc::make_mut(&mut param.data),
             velocity,
             grad,
         )