diff --git a/.ci/scripts/unittest-buck2.sh b/.ci/scripts/unittest-buck2.sh index 295295340fc..e78e682faac 100755 --- a/.ci/scripts/unittest-buck2.sh +++ b/.ci/scripts/unittest-buck2.sh @@ -35,10 +35,17 @@ BUILDABLE_KERNELS_PRIM_OPS_TARGETS=$(buck2 query //kernels/prim_ops/... | grep - for op in "build" "test"; do buck2 $op $BUILDABLE_OPTIMIZED_OPS \ //examples/selective_build:select_all_dtype_selective_lib_portable_lib \ + //extension/llm/custom_ops/spinquant/test:fast_hadamard_transform_test \ + //extension/llm/runner/test:test_multimodal_input \ + //extension/llm/runner/test:test_generation_config \ //kernels/portable/... \ $BUILDABLE_KERNELS_PRIM_OPS_TARGETS //runtime/backend/... //runtime/core/... \ //runtime/executor: //runtime/kernel/... //runtime/platform/... done # Build only without testing -buck2 build //codegen/tools/... # Needs torch for testing which we don't have in our OSS buck setup. +buck2 build //codegen/tools/... \ + //extension/llm/runner/io_manager:io_manager \ + //extension/llm/modules/... \ + //extension/llm/runner:multimodal_runner_lib \ + //extension/llm/runner:text_decoder_runner diff --git a/.lintrunner.toml b/.lintrunner.toml index 0b6a6eb8908..ef771bdb9df 100644 --- a/.lintrunner.toml +++ b/.lintrunner.toml @@ -206,6 +206,7 @@ exclude_patterns = [ '**/*.png', '**/*.webp', '**/*.jpeg', + '**/*.mp3', '**/*.mp4', '**/*.pte', '**/*.pth', @@ -216,6 +217,8 @@ exclude_patterns = [ '**/*.jpg', '**/*.jar', '**/*.gif', + 'extension/llm/tokenizers', + 'extension/llm/tokenizers/**', # File contains @generated 'extension/llm/custom_ops/spinquant/fast_hadamard_transform_special.h', 'extension/llm/custom_ops/spinquant/test/fast_hadamard_transform_special_unstrided_cpu.h', diff --git a/extension/llm/tokenizers b/extension/llm/tokenizers index 4ed91cc545e..86a99c99f5b 160000 --- a/extension/llm/tokenizers +++ b/extension/llm/tokenizers @@ -1 +1 @@ -Subproject commit 4ed91cc545e9ed7098e53747656eb7eff24eb305 +Subproject commit 86a99c99f5b53dc9f01b2a39f0514f91be9bd955