From 3e6468b0976dbfe329b9f9511e770f8ad4092fe1 Mon Sep 17 00:00:00 2001 From: xaedes Date: Mon, 14 Aug 2023 20:56:03 +0200 Subject: [PATCH] fix test when to create temporary backward graph temporary backward graph is only necessary when using checkpointing --- examples/train-text-from-scratch/train-text-from-scratch.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/train-text-from-scratch/train-text-from-scratch.cpp b/examples/train-text-from-scratch/train-text-from-scratch.cpp index a30291a1c..11754ffd9 100644 --- a/examples/train-text-from-scratch/train-text-from-scratch.cpp +++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp @@ -2611,7 +2611,7 @@ int main(int argc, char ** argv) { struct ggml_cgraph * gf = ggml_new_graph(ctx0); struct ggml_cgraph * gb = ggml_new_graph(ctx0); - struct ggml_cgraph * gb_tmp = params.use_alloc + struct ggml_cgraph * gb_tmp = params.use_checkpointing ? ggml_new_graph(ctx0) : NULL;