From b3fcb0091f6cd726b7bb6474b7ad0b4e90d400ab Mon Sep 17 00:00:00 2001 From: Donato Capitella Date: Mon, 23 Feb 2026 09:11:30 +0000 Subject: [PATCH] feat: Enhance `find_max_context.py` with Ray cluster support and fix `C10_HIP_CHECK` build error in Dockerfile. --- Dockerfile | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/Dockerfile b/Dockerfile index 615f47b..1a59c9c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -66,6 +66,14 @@ RUN echo "import sys, re" > patch_strix.py && \ echo "txt = re.sub(r'device_name = .*', 'device_name = \"gfx1151\"', txt)" >> patch_strix.py && \ echo "txt += '\n def get_device_name(self, device_id: int = 0) -> str:\n return \"AMD-gfx1151\"\n'" >> patch_strix.py && \ echo "p.write_text(txt)" >> patch_strix.py && \ + # Patch 3: Fix C10_HIP_CHECK undeclared identifier in selective_scan_fwd.hip + echo "p_hip = Path('csrc/mamba/mamba_ssm/selective_scan_fwd.hip')" >> patch_strix.py && \ + echo "if p_hip.exists():" >> patch_strix.py && \ + echo " txt_hip = p_hip.read_text()" >> patch_strix.py && \ + echo " macro_def = '#ifndef C10_HIP_CHECK\n#include \n#ifndef C10_HIP_CHECK\n#define C10_HIP_CHECK(error) if (error != hipSuccess) { abort(); }\n#endif\n#endif\n'" >> patch_strix.py && \ + echo " txt_hip = macro_def + txt_hip" >> patch_strix.py && \ + echo " p_hip.write_text(txt_hip)" >> patch_strix.py && \ + # ----------- echo "print('Successfully patched vLLM for Strix Halo')" >> patch_strix.py && \ python patch_strix.py && \ sed -i 's/gfx1200;gfx1201/gfx1151/' CMakeLists.txt