utensil

← Commits · 3132b7b6

3132b7b6b55487b25330ac2c33868dd9db5d7632
diff --git a/utensil.h b/utensil.h
index 5851acf..45a702b 100644
--- a/utensil.h
+++ b/utensil.h
@@ -1399,6 +1399,7 @@ float ut_cross_entropy(ut_tensor* logits, const int* labels, ut_tensor* grad_in)
       g[labels[b]] -= 1.f / (float)B;
     }
   }
+  if (grad_in) grad_in->dirty_gpu = true;
   return loss / (float)B;
 }