@@ -188,7 +188,7 @@ jobs:
188188 -Prunner=DataflowRunner \
189189 -PpythonVersion=3.10 \
190190 -PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_rightfit_requirements.txt \
191- '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=CPU --enable_dedup=false --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_cpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_cpu' \
191+ '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=CPU --experiments=enable_streaming_rightfitting -- enable_dedup=false --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_cpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_cpu' \
192192 - name : run PyTorch Image Classification EfficientNet-B0 Streaming (Right-fitting Exactly-once) CPU
193193 uses : ./.github/actions/gradle-command-self-hosted-action
194194 timeout-minutes : 180
@@ -199,7 +199,7 @@ jobs:
199199 -Prunner=DataflowRunner \
200200 -PpythonVersion=3.10 \
201201 -PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_rightfit_requirements.txt \
202- '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=CPU --enable_dedup=true --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_once_cpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_once_cpu' \
202+ '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=CPU --experiments=enable_streaming_rightfitting -- enable_dedup=true --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_once_cpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_once_cpu' \
203203 - name : run PyTorch Image Classification EfficientNet-B0 Streaming (Right-fitting) GPU
204204 uses : ./.github/actions/gradle-command-self-hosted-action
205205 timeout-minutes : 180
@@ -210,7 +210,7 @@ jobs:
210210 -Prunner=DataflowRunner \
211211 -PpythonVersion=3.10 \
212212 -PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_rightfit_requirements.txt \
213- '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=GPU --experiments=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver:5xx --enable_dedup=false --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_gpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_gpu' \
213+ '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=GPU --experiments=enable_streaming_rightfitting, worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver:5xx --enable_dedup=false --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_gpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_gpu' \
214214 - name : run PyTorch Image Classification EfficientNet-B0 Streaming (Right-fitting Exactly-once) GPU
215215 uses : ./.github/actions/gradle-command-self-hosted-action
216216 timeout-minutes : 180
@@ -221,4 +221,4 @@ jobs:
221221 -Prunner=DataflowRunner \
222222 -PpythonVersion=3.10 \
223223 -PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_rightfit_requirements.txt \
224- '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=GPU --experiments=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver:5xx --enable_dedup=true --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_once_gpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_once_gpu'
224+ '-PloadTest.args=${{ env.beam_Inference_Python_Benchmarks_Dataflow_2_test_arguments_1 }} --device=GPU --experiments=enable_streaming_rightfitting, worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver:5xx --enable_dedup=true --mode=streaming --job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}} --metrics_table=torch_inference_imagenet_stream_rightfit_once_gpu --output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_once_gpu'
0 commit comments