summaryrefslogtreecommitdiffstats
path: root/source
diff options
context:
space:
mode:
Diffstat (limited to 'source')
-rw-r--r--source/slang/slang-diagnostic-defs.h6
-rw-r--r--source/slang/slang-emit-c-like.cpp18
-rw-r--r--source/slang/slang-emit-c-like.h3
3 files changed, 26 insertions, 1 deletions
diff --git a/source/slang/slang-diagnostic-defs.h b/source/slang/slang-diagnostic-defs.h
index dbd4da588..808d867b6 100644
--- a/source/slang/slang-diagnostic-defs.h
+++ b/source/slang/slang-diagnostic-defs.h
@@ -2765,6 +2765,12 @@ DIAGNOSTIC(
noBlocksOrIntrinsic,
"no blocks found for function definition, is there a '$0' intrinsic missing?")
+DIAGNOSTIC(
+ 40100,
+ Warning,
+ mainEntryPointRenamed,
+ "entry point '$0' is not allowed, and has been renamed to '$1'")
+
//
// Ray tracing
//
diff --git a/source/slang/slang-emit-c-like.cpp b/source/slang/slang-emit-c-like.cpp
index 141a843f2..b2dc9b014 100644
--- a/source/slang/slang-emit-c-like.cpp
+++ b/source/slang/slang-emit-c-like.cpp
@@ -1109,9 +1109,25 @@ void CLikeSourceEmitter::appendScrubbedName(const UnownedStringSlice& name, Stri
}
}
+inline String CLikeSourceEmitter::maybeMakeEntryPointNameValid(String name, DiagnosticSink* sink)
+{
+ if (isCPUTarget(getTargetReq()) || isCUDATarget(getTargetReq()) ||
+ isMetalTarget(getTargetReq()))
+ {
+ if (name == "main")
+ {
+ String newName = _generateUniqueName(name.getUnownedSlice());
+ sink->diagnose(SourceLoc(), Diagnostics::mainEntryPointRenamed, name, newName);
+ return newName;
+ }
+ }
+ return name;
+}
+
String CLikeSourceEmitter::generateEntryPointNameImpl(IREntryPointDecoration* entryPointDecor)
{
- return entryPointDecor->getName()->getStringSlice();
+ String name = entryPointDecor->getName()->getStringSlice();
+ return maybeMakeEntryPointNameValid(name, getSink());
}
String CLikeSourceEmitter::_generateUniqueName(const UnownedStringSlice& name)
diff --git a/source/slang/slang-emit-c-like.h b/source/slang/slang-emit-c-like.h
index f38158c0a..78793f655 100644
--- a/source/slang/slang-emit-c-like.h
+++ b/source/slang/slang-emit-c-like.h
@@ -739,6 +739,9 @@ protected:
Dictionary<const char*, IRStringLit*> m_builtinPreludes;
+ // Rename entry point if target doesn't allow the name (e.g., 'main')
+ virtual String maybeMakeEntryPointNameValid(String name, DiagnosticSink* sink);
+
// Indicates if we are emiting for DXC cooperative vector POC.
bool isCoopvecPoc = false;
};