[V1] Use custom ops for piecewise CUDA graphs (#10227)
Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
parent
330e82d34a
commit
fe15729a2b
@ -1,4 +1,3 @@
|
||||
import os
|
||||
import time
|
||||
from dataclasses import dataclass
|
||||
from typing import TYPE_CHECKING, Dict, List, Optional, Set
|
||||
@ -406,7 +405,6 @@ class GPUModelRunner:
|
||||
if self.use_cuda_graph:
|
||||
# FIXME(woosuk): Currently, we do not use inductor to reduce the
|
||||
# compilation time and any potential issues with the inductor.
|
||||
os.environ["VLLM_CUSTOM_OPS"] = "none"
|
||||
set_compilation_config(
|
||||
CompilationConfig(
|
||||
use_cudagraph=True,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user