From 259a015feb9d4ab65e8fbba32f6c777e92780cc7 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Wed, 22 Mar 2023 21:16:35 -0700
Subject: Type legalization and autodiff bug fixes. (#2722)

* Bug fixes.

* Fix.

* Only perform autodiff for functions whose derivative is actually used.

* Fix loop optimize bug.

* Fix high order diff.

* Fix trivial diff func generation.

* Fixes.

* Cleanup.

---------

Co-authored-by: Yong He <yhe@nvidia.com>
---
 source/slang/diff.meta.slang | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

(limited to 'source/slang/diff.meta.slang')

diff --git a/source/slang/diff.meta.slang b/source/slang/diff.meta.slang
index a9b8209f3..26a673512 100644
--- a/source/slang/diff.meta.slang
+++ b/source/slang/diff.meta.slang
@@ -1022,31 +1022,35 @@ __generic<T : __BuiltinFloatingPointType, let N : int>
 [__readNone]
 T __determinant_impl(matrix<T,N,N> m)
 {
-    if (N == 1)
-        return m[0][0];
-    else if (N == 2)
-        return m[0][0] * m[1][1] - m[0][1] * m[1][0];
-    else if (N == 3)
+    T result = T(0);
+    switch (N)
     {
-        return  m[0][0] * (m[1][1] * m[2][2] - m[2][1] * m[1][2])
+    case 1:
+        result = m[0][0];
+        break;
+    case 2:
+        result = m[0][0] * m[1][1] - m[0][1] * m[1][0];
+        break;
+    case 3:
+        result =  m[0][0] * (m[1][1] * m[2][2] - m[2][1] * m[1][2])
 		      - m[1][0] * (m[0][1] * m[2][2] - m[2][1] * m[0][2])
 			  + m[2][0] * (m[0][1] * m[1][2] - m[1][1] * m[0][2]);
-    }
-    else if (N == 4)
-    {
-    	T s00 = m[2][2] * m[3][3] - m[3][2] * m[2][3];
+        break;
+    case 4:
+        T s00 = m[2][2] * m[3][3] - m[3][2] * m[2][3];
 		T s01 = m[2][1] * m[3][3] - m[3][1] * m[2][3];
 		T s02 = m[2][1] * m[3][2] - m[3][1] * m[2][2];
 		T s03 = m[2][0] * m[3][3] - m[3][0] * m[2][3];
 		T s04 = m[2][0] * m[3][2] - m[3][0] * m[2][2];
 		T s05 = m[2][0] * m[3][1] - m[3][0] * m[2][1];
 
-		return m[0][0] * (m[1][1] * s00 - m[1][2] * s01 + m[1][3] * s02)
+		result = m[0][0] * (m[1][1] * s00 - m[1][2] * s01 + m[1][3] * s02)
 			 - m[0][1] * (m[1][0] * s00 - m[1][2] * s03 + m[1][3] * s04)
 			 + m[0][2] * (m[1][0] * s01 - m[1][1] * s03 + m[1][3] * s05)
 			 - m[0][3] * (m[1][0] * s02 - m[1][1] * s04 + m[1][2] * s05);
+        break;
     }
-    return T(0.0);
+    return result;
 }
 __generic<T : __BuiltinFloatingPointType, let N : int>
 [ForwardDerivativeOf(determinant)]
-- 
cgit v1.2.3