From 7c8c6cd67f6ab7c8cbceb037938fe3093a90bb7b Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Mon, 11 May 2026 06:06:52 +0000
Subject: [PATCH] fix: add PYTHONPATH for deep_gemm import

---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile b/Dockerfile
index 6f5e25a..0277f24 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,6 +12,7 @@ RUN git clone -b nvfp4-mega-moe https://sweetapi.com/biondizzle/DeepGEMM.git /ro
 
 # Build DeepGEMM with proper CUDA/NVRTC paths
 ENV CPATH="/usr/local/lib/python3.12/dist-packages/flashinfer/data/cutlass/include:/usr/local/lib/python3.12/dist-packages/nvidia/cu13/include:/usr/local/cuda-13.0/include:${CPATH}"
+ENV PYTHONPATH="/root/DeepGEMM:${PYTHONPATH}"
 # NVRTC lives in the pip nvidia/cu13 package, but the linker expects it in cuda/lib64
 # Create a symlink so -lnvrtc resolves
 RUN ln -sf /usr/local/lib/python3.12/dist-packages/nvidia/cu13/lib/libnvrtc.so.13 /usr/local/cuda/lib64/libnvrtc.so