diff options
Diffstat (limited to 'candle-examples/examples/llama_multiprocess/model.rs')
| -rw-r--r-- | candle-examples/examples/llama_multiprocess/model.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/llama_multiprocess/model.rs b/candle-examples/examples/llama_multiprocess/model.rs index ab4e382c..ad5e4cd2 100644 --- a/candle-examples/examples/llama_multiprocess/model.rs +++ b/candle-examples/examples/llama_multiprocess/model.rs @@ -68,7 +68,7 @@ impl CustomOp1 for AllReduce { } fn all_reduce_sum(x: &Tensor, comm: &Rc<Comm>) -> Result<Tensor> { - x.custom_op1(AllReduce { comm: comm.clone() }) + x.apply_op1(AllReduce { comm: comm.clone() }) } impl TensorParallelRowLinear { |