Преглед на файлове

r36 build: update b12x gate marker (B12X_PCIE_ONESHOT_PDL removed in b12x 323107ff; use _PREPARED_ONESHOT_LAUNCHERS — same file, r36 oneshot launcher tables)

Alexander Korolev преди 1 месец
родител
ревизия
09a88f2ced
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      docker-images/glm53-jovian-r27/Dockerfile

+ 1 - 1
docker-images/glm53-jovian-r27/Dockerfile

@@ -271,7 +271,7 @@ RUN set -eux; \
 assert E.B12X.value and 'b12x' in E.B12X.value.lower(); print('B12X enum ok')"; \
     grep -q "AttentionBackendEnum.B12X," /usr/local/lib/python3.12/dist-packages/vllm/platforms/cuda.py; \
     grep -Fq 'def apply_top_k_top_p_probs' /usr/local/lib/python3.12/dist-packages/vllm/v1/sample/ops/topk_topp_sampler.py; \
-    grep -Fq 'B12X_PCIE_ONESHOT_PDL' /usr/local/lib/python3.12/dist-packages/b12x/comm/pcie/_oneshot_cute.py; \
+    grep -Fq '_PREPARED_ONESHOT_LAUNCHERS' /usr/local/lib/python3.12/dist-packages/b12x/comm/pcie/_oneshot_cute.py; \
     grep -Fq 'self._transfer_workspace_slot_count = self._commit_workers' /usr/local/lib/python3.12/dist-packages/lmcache/v1/multiprocess/transfer_context/async_engine_driven.py; \
     test -s /opt/lmcache/lib/liblmcache_cumem_shareable.so