Skip to content

Commit 83894ea

Browse files
committed
Fix api_stability unittest
Signed-off-by: Dan Hansen <[email protected]>
1 parent 1137bda commit 83894ea

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

tensorrt_llm/llmapi/llm_args.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2573,7 +2573,7 @@ class TorchLlmArgs(BaseLlmArgs):
25732573
sampler_enable_async_worker: bool = Field(
25742574
default=False,
25752575
description="Enable the async worker in the sampler for D->H copies",
2576-
status="beta")
2576+
status="prototype")
25772577

25782578
enable_iter_perf_stats: bool = Field(
25792579
default=False,

tests/unittest/api_stability/references/llm.yaml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,10 @@ methods:
119119
annotation: Union[str, tensorrt_llm.llmapi.llm_args.SamplerType]
120120
default: auto
121121
status: beta
122+
sampler_enable_async_worker:
123+
annotation: bool
124+
default: False
125+
status: prototype
122126
enable_iter_perf_stats:
123127
annotation: bool
124128
default: False

0 commit comments

Comments
 (0)