mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-07-20 14:57:16 +08:00
init
Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
parent
1fc9de4c1a
commit
2f29120a09
@ -340,7 +340,6 @@ class UBatchWrapper:
|
|||||||
if cudagraph_runtime_mode is CUDAGraphMode.FULL:
|
if cudagraph_runtime_mode is CUDAGraphMode.FULL:
|
||||||
assert batch_descriptor is not None
|
assert batch_descriptor is not None
|
||||||
if batch_descriptor.num_tokens in self.cudagraphs:
|
if batch_descriptor.num_tokens in self.cudagraphs:
|
||||||
logger.info("DISABLEING CUDAGRAPHS")
|
|
||||||
cudagraph_runtime_mode = CUDAGraphMode.NONE
|
cudagraph_runtime_mode = CUDAGraphMode.NONE
|
||||||
|
|
||||||
if cudagraph_runtime_mode in (CUDAGraphMode.NONE,
|
if cudagraph_runtime_mode in (CUDAGraphMode.NONE,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user