more windows gumph

johnhaddon · Nov 15, 2024 · 6e382f7 · 6e382f7
1 parent 565562c
commit 6e382f7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/bin/gaffer.cmd b/bin/gaffer.cmd
@@ -140,6 +140,12 @@ if "%CYCLES_ROOT%" NEQ "" (
 	call :prependToPath "%CYCLES_ROOT%\bin" PATH
 )
 
+rem ONNX
+rem ====
+
+if "%ONNX_ROOT%" NEQ "" (
+	call :appendToPath "%ONNX_ROOT%\lib" PATH
+)
 
 rem Set up 3rd party extensions
 rem Batch files are awkward at `for` loops. The default `for`, without `/f`

diff --git a/python/GafferML/__init__.py b/python/GafferML/__init__.py
@@ -34,9 +34,16 @@
 #
 ##########################################################################
 
+import os
+import pathlib
+
 __import__( "Gaffer" )
 __import__( "GafferImage" )
 
+if hasattr( os, "add_dll_directory" ) :
+	os.add_dll_directory( ( pathlib.Path( os.environ["ONNX_ROOT"] ) / "lib" ).resolve() )
+del os, pathlib # Don't pollute the namespace
+
 from ._GafferML import *
 
 __import__( "IECore" ).loadConfig( "GAFFER_STARTUP_PATHS", subdirectory = "GafferML" )