From 3bc6ad98be76a5266122bd8ce965ee45a8fa682d Mon Sep 17 00:00:00 2001
From: Tarry Singh <tarry.singh@deepkapha.com>
Date: Fri, 6 Oct 2023 10:10:56 +0000
Subject: [PATCH] Update module.zig to donate all buffers except the
 `token_index` buffer for the Llama+Neuron example.

---
 zml/module.zig | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/zml/module.zig b/zml/module.zig
index 5ced279..e2c621c 100644
--- a/zml/module.zig
+++ b/zml/module.zig
@@ -350,9 +350,6 @@ pub const CompilationContext = struct {
     /// Given a list of donations mapping output buffers to input buffers,
     /// generate donation attribute for each `n_args` input argument.
     fn addDonationsAttributes(self: CompilationContext, attributes: []AttributeList, donations: []const Tensor._Donation) void {
-        if (self.target() == .neuron) {
-            return;
-        }
         var n_donations: usize = 0;
         for (donations, 0..) |donation, index| {
             switch (donation) {