Merge pull request #1936 from KhronosGroup/rt-stage-cli

Add --stage handling for ray tracing.

1 files changed, 12 insertions, 0 deletions

diff --git a/main.cpp b/main.cpp
index c4ff4c4e..521e98f9 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1070,6 +1070,18 @@ static ExecutionModel stage_to_execution_model(const std::string &stage)
 		return ExecutionModelTessellationEvaluation;
 	else if (stage == "geom")
 		return ExecutionModelGeometry;
+	else if (stage == "rgen")
+		return ExecutionModelRayGenerationKHR;
+	else if (stage == "rint")
+		return ExecutionModelIntersectionKHR;
+	else if (stage == "rahit")
+		return ExecutionModelAnyHitKHR;
+	else if (stage == "rchit")
+		return ExecutionModelClosestHitKHR;
+	else if (stage == "rmiss")
+		return ExecutionModelMissKHR;
+	else if (stage == "rcall")
+		return ExecutionModelCallableKHR;
 	else
 		SPIRV_CROSS_THROW("Invalid stage.");
 }