Objective-C interop + a few improvements

1 files changed, 8 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index b6392cc..f5d8c1b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -172,6 +172,10 @@ int main(int argc, char **argv)
             g_config.use_cuda = 1;
             g_config.use_cpp = 1; // CUDA implies C++ mode.
         }
+        else if (strcmp(arg, "--objc") == 0)
+        {
+            g_config.use_objc = 1;
+        }
         else if (strcmp(arg, "--check") == 0)
         {
             g_config.mode_check = 1;
@@ -304,6 +308,10 @@ int main(int argc, char **argv)
     {
         temp_source_file = "out.cpp";
     }
+    else if (g_config.use_objc)
+    {
+        temp_source_file = "out.m";
+    }
 
     // Codegen to C/C++/CUDA
     FILE *out = fopen(temp_source_file, "w");