Optional startup flags to override performance-related defaults for the compute shaders

1 files changed, 6 insertions, 4 deletions

diff --git a/Whisper/modelFactory.cpp b/Whisper/modelFactory.cpp
index a708551..d50b854 100644
--- a/Whisper/modelFactory.cpp
+++ b/Whisper/modelFactory.cpp
@@ -1,16 +1,18 @@
-#include "stdafx.h"
+#include "stdafx.h"
 #include "modelFactory.h"
 #include "API/iContext.cl.h"
 
-HRESULT COMLIGHTCALL Whisper::loadModel( const wchar_t* path, eModelImplementation impl, const sLoadModelCallbacks* callbacks, iModel** pp )
+HRESULT COMLIGHTCALL Whisper::loadModel( const wchar_t* path, eModelImplementation impl, uint32_t flags, const sLoadModelCallbacks* callbacks, iModel** pp )
 {
 	switch( impl )
 	{
 	case eModelImplementation::GPU:
-		return loadGpuModel( path, false, callbacks, pp );
+		return loadGpuModel( path, false, flags, callbacks, pp );
 	case eModelImplementation::Hybrid:
-		return loadGpuModel( path, true, callbacks, pp );
+		return loadGpuModel( path, true, flags, callbacks, pp );
 	case eModelImplementation::Reference:
+		if( 0 != flags )
+			logWarning( u8"The reference model doesn’t currently use any flags, argument ignored" );
 		return loadReferenceCpuModel( path, pp );
 	}