accepts ptx extension when target is nvptx64

2013-10-29 10:21:48 +01:00
parent ff98271a43
commit 57aefdf830
1 changed files with 7 additions and 1 deletions
--- a/module.cpp
+++ b/module.cpp
@@ -953,7 +953,13 @@ Module::writeOutput(OutputType outputType, const char *outFileName,
        const char *fileType = NULL;
        switch (outputType) {
        case Asm:
-            if (strcasecmp(suffix, "s"))
+            if (g->target->getISA() != Target::NVPTX64)
+            {
+              if (strcasecmp(suffix, "s"))
+                fileType = "assembly";
+            }
+            else
+              if (strcasecmp(suffix, "ptx"))
                fileType = "assembly";
            break;
        case Bitcode: