From 30b5335fed9524101d980618b0749dcaf546c319 Mon Sep 17 00:00:00 2001
From: winmad <winmad.wlf@gmail.com>
Date: Fri, 2 Dec 2022 01:01:58 -0800
Subject: Add helper functions to update DifferentialPair (#2547)

---
 source/slang/diff.meta.slang | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'source')

diff --git a/source/slang/diff.meta.slang b/source/slang/diff.meta.slang
index 033c173ab..a97ab9eaf 100644
--- a/source/slang/diff.meta.slang
+++ b/source/slang/diff.meta.slang
@@ -92,6 +92,24 @@ struct DifferentialPair : IDifferentiable
     }
 };
 
+[ForceInline]
+void updatePrimal<T : IDifferentiable>(inout DifferentialPair<T> p, T newPrimal)
+{
+    p = DifferentialPair<T>(newPrimal, p.d);
+}
+
+[ForceInline]
+void updateDiff<T : IDifferentiable>(inout DifferentialPair<T> p, T.Differential newDiff)
+{
+    p = DifferentialPair<T>(p.p, newDiff);
+}
+
+[ForceInline]
+void updatePair<T : IDifferentiable>(inout DifferentialPair<T> p, T newPrimal, T.Differential newDiff)
+{
+    p = DifferentialPair<T>(newPrimal, newDiff);
+}
+
 
 #define VECTOR_MAP_D_UNARY(TYPE, COUNT, D_FUNC, VALUE) \
     vector<TYPE, COUNT> result; \
-- 
cgit v1.2.3