fix

Author: WANDY666

Dockerfile

@@ -82,7 +82,7 @@ RUN echo "🔧 Building lightllm-kernel package..." && \
 # Build flash_attn_3 package (hopper)
 RUN echo "🔧 Building flash_attn_3 package..." && \
     cd flash-attention/hopper && \
-    MAX_JOBS=2 NVCC_THREADS=2 FLASH_ATTN_CUDA_ARCHS=90 python setup.py bdist_wheel && \
+    MAX_JOBS=2 NVCC_THREADS=2 FLASH_ATTN_CUDA_ARCHS=90 FLASH_ATTENTION_DISABLE_SM80=TRUE python setup.py bdist_wheel && \
     cp dist/*.whl /out/ && \
     echo "✅ flash_attn_3 build completed"
 
@@ -95,4 +95,4 @@ RUN echo "📦 Final wheel packages:" && \
         echo "❌ Error: Expected 2 wheels, found $WHEEL_COUNT" && exit 1; \
     else \
         echo "✅ Successfully built all wheel packages"; \
-    fi 
+    fi