Pass WriteToOutput to backends. (#1066)

* Added more code-generation functions to IBackendCodeGenerator. * Adapted Backends to updated IBackendCodeGenerator.
m4rs-mt · Aug 31, 2023 · 89929e8 · 89929e8
1 parent a2461fa
commit 89929e8
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/Src/ILGPU/Backends/IBackendCodeGenerator.cs b/Src/ILGPU/Backends/IBackendCodeGenerator.cs
@@ -1,6 +1,6 @@
 // ---------------------------------------------------------------------------------------
 //                                        ILGPU
-//                        Copyright (c) 2020-2021 ILGPU Project
+//                        Copyright (c) 2020-2023 ILGPU Project
 //                                    www.ilgpu.net
 //
 // File: IBackendCodeGenerator.cs
@@ -317,6 +317,12 @@ public interface IBackendCodeGenerator
         /// <param name="debug">The node.</param>
         void GenerateCode(DebugAssertOperation debug);
 
+        /// <summary>
+        /// Generates code for the output writer.
+        /// </summary>
+        /// <param name="writeToOutput">The node.</param>
+        void GenerateCode(WriteToOutput writeToOutput);
+
         // Terminators
 
         /// <summary>
@@ -594,7 +600,7 @@ public void Visit(DebugAssertOperation debug) =>
 
             /// <summary cref="IValueVisitor.Visit(WriteToOutput)"/>
             public void Visit(WriteToOutput writeToOutput) =>
-                throw new InvalidCodeGenerationException();
+                CodeGenerator.GenerateCode(writeToOutput);
 
             /// <summary cref="IValueVisitor.Visit(ReturnTerminator)"/>
             public void Visit(ReturnTerminator returnTerminator) =>

diff --git a/Src/ILGPU/Backends/OpenCL/CLCodeGenerator.Values.cs b/Src/ILGPU/Backends/OpenCL/CLCodeGenerator.Values.cs
@@ -792,6 +792,11 @@ public void GenerateCode(DebugAssertOperation debug) =>
             // Invalid debug node -> should have been removed
             debug.Assert(false);
 
+        /// <summary cref="IBackendCodeGenerator.GenerateCode(WriteToOutput)"/>
+        public void GenerateCode(WriteToOutput writeToOutput) =>
+            // Invalid write node -> should have been removed
+            writeToOutput.Assert(false);
+
         /// <summary cref="IBackendCodeGenerator.GenerateCode(LanguageEmitValue)"/>
         public void GenerateCode(LanguageEmitValue value) =>
             // Ignore PTX instructions.

diff --git a/Src/ILGPU/Backends/PTX/PTXCodeGenerator.Values.cs b/Src/ILGPU/Backends/PTX/PTXCodeGenerator.Values.cs
@@ -1231,7 +1231,13 @@ public void GenerateCode(SubWarpShuffle shuffle)
 
         /// <summary cref="IBackendCodeGenerator.GenerateCode(DebugAssertOperation)"/>
         public void GenerateCode(DebugAssertOperation debug) =>
-            Debug.Assert(false, "Invalid debug node -> should have been removed");
+            // Invalid debug node -> should have been removed
+            debug.Assert(false);
+
+        /// <summary cref="IBackendCodeGenerator.GenerateCode(WriteToOutput)"/>
+        public void GenerateCode(WriteToOutput writeToOutput) =>
+            // Invalid write node -> should have been removed
+            writeToOutput.Assert(false);
 
         /// <summary cref="IBackendCodeGenerator.GenerateCode(LanguageEmitValue)"/>
         [SuppressMessage(