This is an automated email from the ASF dual-hosted git repository. vterentev pushed a commit to branch oss-image-cpu in repository https://gitbox.apache.org/repos/asf/beam.git
commit f5c8a7eea1f290523f7649c5d2649e43c13a9407 Author: Vitaly Terentyev <[email protected]> AuthorDate: Wed Jan 21 22:01:10 2026 +0400 Test oss image only --- .../beam_Inference_Python_Benchmarks_Dataflow.yml | 95 ---------------------- 1 file changed, 95 deletions(-) diff --git a/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml b/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml index edd92bec660..883f5dc4457 100644 --- a/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml +++ b/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml @@ -98,101 +98,6 @@ jobs: # The env variables are created and populated in the test-arguments-action as "<github.job>_test_arguments_<argument_file_paths_index>" - name: get current time run: echo "NOW_UTC=$(date '+%m%d%H%M%S' --utc)" >> $GITHUB_ENV - - name: Build VLLM Development Image - id: build_vllm_image - uses: ./.github/actions/build-push-docker-action - with: - dockerfile_path: 'sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile' - image_name: 'us-docker.pkg.dev/apache-beam-testing/beam-temp/beam-vllm-gpu-base' - image_tag: ${{ github.sha }} - - name: Run VLLM Gemma Batch Test - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.vllm_gemma_benchmarks \ - -Prunner=DataflowRunner \ - -PsdkLocationOverride=false \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/vllm_tests_requirements.txt '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_8 }} --mode=batch --job_name=benchmark-tests-vllm-with-gemma-2b-it-batch-${{env.NOW_UTC}} --sdk_container_image=${{ steps.build_vllm_image.outputs.image_url }}' - - name: run Pytorch Sentiment Streaming using Hugging Face distilbert-base-uncased model - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_sentiment_benchmarks \ - -Prunner=DataflowRunner \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/torch_tests_requirements.txt \ - '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_6 }} --mode=streaming --job_name=benchmark-tests-pytorch-sentiment-distilbert-base-uncased-streaming-${{env.NOW_UTC}} --output_table=apache-beam-testing.beam_run_inference.result_sentiment_distilbert_base_uncased' \ - - name: run Pytorch Sentiment Batch using Hugging Face distilbert-base-uncased model - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_sentiment_benchmarks \ - -Prunner=DataflowRunner \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/torch_tests_requirements.txt \ - '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_7 }} --mode=batch --job_name=benchmark-tests-pytorch-sentiment-distilbert-base-uncased-batch-${{env.NOW_UTC}} --output_table=apache-beam-testing.beam_run_inference.result_sentiment_distilbert_base_uncased' \ - - name: run Pytorch Vision Classification with Resnet 101 - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_image_classification_benchmarks \ - -Prunner=DataflowRunner \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/torch_tests_requirements.txt \ - '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_1 }} --job_name=benchmark-tests-pytorch-imagenet-python-101-${{env.NOW_UTC}} --output=gs://temp-storage-for-end-to-end-tests/torch/result_resnet101-${{env.NOW_UTC}}.txt' \ - - name: run Pytorch Imagenet Classification with Resnet 152 - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_image_classification_benchmarks \ - -Prunner=DataflowRunner \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/torch_tests_requirements.txt \ - '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_2 }} --job_name=benchmark-tests-pytorch-imagenet-python-152-${{env.NOW_UTC}} --output=gs://temp-storage-for-end-to-end-tests/torch/result_resnet152-${{env.NOW_UTC}}.txt' \ - - name: run Pytorch Language Modeling using Hugging Face bert-base-uncased model - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_language_modeling_benchmarks \ - -Prunner=DataflowRunner \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/torch_tests_requirements.txt \ - '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_3 }} --job_name=benchmark-tests-pytorch-language-modeling-bert-base-uncased-${{env.NOW_UTC}} --output=gs://temp-storage-for-end-to-end-tests/torch/result_bert_base_uncased-${{env.NOW_UTC}}.txt' \ - - name: run Pytorch Langauge Modeling using Hugging Face bert-large-uncased model - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_language_modeling_benchmarks \ - -Prunner=DataflowRunner \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/torch_tests_requirements.txt \ - '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_4 }} --job_name=benchmark-tests-pytorch-language-modeling-bert-large-uncased-${{env.NOW_UTC}} --output=gs://temp-storage-for-end-to-end-tests/torch/result_bert_large_uncased-${{env.NOW_UTC}}.txt' \ - - name: run Pytorch Imagenet Classification with Resnet 152 with Tesla T4 GPU - uses: ./.github/actions/gradle-command-self-hosted-action - timeout-minutes: 180 - with: - gradle-command: :sdks:python:apache_beam:testing:load_tests:run - arguments: | - -PloadTest.mainClass=apache_beam.testing.benchmarks.inference.pytorch_image_classification_benchmarks \ - -Prunner=DataflowRunner \ - -PpythonVersion=3.10 \ - -PloadTest.requirementsTxtFile=apache_beam/ml/inference/torch_tests_requirements.txt \ - '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_5 }} --job_name=benchmark-tests-pytorch-imagenet-python-gpu-${{env.NOW_UTC}} --output=gs://temp-storage-for-end-to-end-tests/torch/result_resnet152_gpu-${{env.NOW_UTC}}.txt' \ - name: run PyTorch Image Classification EfficientNet-B0 Streaming (Right-fitting) uses: ./.github/actions/gradle-command-self-hosted-action timeout-minutes: 180
