From 191da4ee060efbe0d7771c7459a9aacaa663fd83 Mon Sep 17 00:00:00 2001 From: PGZXB Date: Mon, 7 Nov 2022 12:35:27 +0800 Subject: [PATCH] [test] Test real function on enabling the offline cache (#6523) Issue: #4401 Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- tests/python/test_offline_cache.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/test_offline_cache.py b/tests/python/test_offline_cache.py index 522bfb685418f..f80b1b38ba8c5 100644 --- a/tests/python/test_offline_cache.py +++ b/tests/python/test_offline_cache.py @@ -548,8 +548,8 @@ def added_files(arch): # FIXME: Change to `supported_archs_offline_cache` after fixing bugs of real-function on gpu -@pytest.mark.parametrize( - 'curr_arch', [ti.cpu] if ti.cpu in test_utils.expected_archs() else []) +@pytest.mark.parametrize('curr_arch', + {ti.cpu, ti.cuda} & supported_archs_offline_cache) @_test_offline_cache_dec def test_offline_cache_for_kernels_calling_real_func(curr_arch): count_of_cache_file = cache_files_cnt(curr_arch)