Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions tests/integration/test_lists/test-db/l0_a10.yml
Original file line number Diff line number Diff line change
Expand Up @@ -245,6 +245,7 @@ l0_a10:
- llmapi/test_llm_examples.py::test_llmapi_quickstart_atexit
- unittest/test_model_runner_cpp.py
- unittest/llmapi/test_build_cache.py
- unittest/others/test_builder.py
- unittest/trt/attention/test_gpt_attention_IFB.py
- unittest/trt/attention/test_gpt_attention_no_cache.py
- condition:
Expand Down
4 changes: 3 additions & 1 deletion tests/unittest/others/test_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,9 @@ def test_basic_builder_flow(self):
assert engine is not None
refit_engine = builder.refit_engine(network, engine)
assert refit_engine is not None
builder.save_config(builder_config, tempfile.mktemp())
with tempfile.TemporaryDirectory() as tmpdir:
config_path = f"{tmpdir}/builder_config.json"
builder.save_config(builder_config, config_path)

def test_top_level_dont_have_functional_apis(self):
# This did not check all the functional apis, but should already prevent
Expand Down
Loading