DEBUG stuff

Author: NicolasHug

.github/workflows/windows_wheel.yaml

@@ -183,6 +183,7 @@ jobs:
         export GPU_ARCH_TYPE=${{ matrix.gpu-arch-type }}
         export GPU_ARCH_VERSION=${{ matrix.gpu-arch-version }}
         export FFMPEG_VERSION_FOR_TESTS=${{ matrix.ffmpeg-version-for-tests }}
+        export BUILD_AGAINST_ALL_FFMPEG_FROM_S3=1
 
         # Prepare conda
         set +x && eval "$($(which conda) shell.bash hook)" && set -x
@@ -213,8 +214,32 @@ jobs:
         # Install torchcodec from source in development mode
         pip install -e ".[dev]" --no-build-isolation -vv
         
-        # Test torchcodec import
-        python -c "import torchcodec; print('TorchCodec import successful!')"
+        # Add torchcodec source directory to DLL search path for Windows
+        torchcodec_dll_path="$(pwd)/src/torchcodec"
+        echo "Adding torchcodec DLL path: $torchcodec_dll_path"
+        export PATH="$torchcodec_dll_path:$PATH"
+        
+        # Test torchcodec import with detailed error reporting
+        python -c "
+import torch
+print(f'PyTorch version: {torch.__version__}')
+
+try:
+    import torchcodec
+    print('TorchCodec import successful!')
+except Exception as e:
+    print(f'TorchCodec import failed: {e}')
+    import traceback
+    traceback.print_exc()
+    
+    # Try to get more specific error info
+    try:
+        from torchcodec._core import ops
+        print('Core ops import successful')
+    except Exception as ops_e:
+        print(f'Core ops import failed: {ops_e}')
+        traceback.print_exc()
+"
         
         # Run tests
         pytest test -vvv

src/torchcodec/_core/ops.py

@@ -58,6 +58,16 @@ def load_torchcodec_shared_libraries():
             if sys.platform == "win32":
                 try:
                     import ctypes
+                    
+                    # Add the DLL directory to help with dependency resolution
+                    dll_dir = os.path.dirname(core_path)
+                    if hasattr(os, 'add_dll_directory'):
+                        try:
+                            os.add_dll_directory(dll_dir)
+                            print(f"Added DLL directory: {dll_dir}", flush=True)
+                        except Exception as add_dir_e:
+                            print(f"Could not add DLL directory: {add_dir_e}", flush=True)
+                    
                     # Preload the core library so it's available for dependency resolution
                     ctypes.CDLL(core_path)
                     print(f"Preloaded core library via ctypes", flush=True)