Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor ExecuteReplicated to operate on sharded data directly #5737

Merged
merged 24 commits into from
Nov 30, 2023

fix merge

674f2c3
Select commit
Loading
Failed to load commit list.
Merged

Refactor ExecuteReplicated to operate on sharded data directly #5737

fix merge
674f2c3
Select commit
Loading
Failed to load commit list.
Google Cloud Build / ci-tpu-test-trigger (tpu-pytorch) succeeded Nov 30, 2023 in 1h 11m 48s

Summary

Build Information

Trigger ci-tpu-test-trigger
Build ca3303d1-656b-40dc-ab0b-d752147b147f
Start 2023-11-30T11:11:15-08:00
Duration 1h11m1.627s
Status SUCCESS

Steps

Step Status Duration
git_fetch SUCCESS 9.525s
git_checkout SUCCESS 619ms
build_pytorch-xla-test_docker_image SUCCESS 53m45.446s
push_pytorch-xla-test_docker_image SUCCESS 1m3.226s
run_e2e_tests SUCCESS 15m59.003s

Details

starting build "ca3303d1-656b-40dc-ab0b-d752147b147f"

FETCHSOURCE
hint: Using 'master' as the name for the initial branch. This default branch name
hint: is subject to change. To configure the initial branch name to use in all
hint: of your new repositories, which will suppress this warning, call:
hint: 
hint: 	git config --global init.defaultBranch <name>
hint: 
hint: Names commonly chosen instead of 'master' are 'main', 'trunk' and
hint: 'development'. The just-created branch can be renamed via this command:
hint: 
hint: 	git branch -m <name>
Initialized empty Git repository in /workspace/.git/
From https://github.com/pytorch/xla
 * branch            674f2c3839b2e1280b0910cc87ab925a45aeb343 -> FETCH_HEAD
HEAD is now at 674f2c3 fix merge
BUILD
Starting Step #0 - "git_fetch"
Step #0 - "git_fetch": Already have image (with digest): gcr.io/cloud-builders/git
Step #0 - "git_fetch": From https://github.com/pytorch/xla
Step #0 - "git_fetch":  * [new branch]          2-0-python-3-10         -> origin/2-0-python-3-10
Step #0 - "git_fetch":  * [new branch]          4483-improve-resnet50-data-loading-performance -> origin/4483-improve-resnet50-data-loading-performance
Step #0 - "git_fetch":  * [new branch]          5d-unit-test            -> origin/5d-unit-test
Step #0 - "git_fetch":  * [new branch]          FixTransferToServer     -> origin/FixTransferToServer
Step #0 - "git_fetch":  * [new branch]          JackCaaoG/spmd_device_python_test -> origin/JackCaaoG/spmd_device_python_test
Step #0 - "git_fetch":  * [new branch]          JackCao/Disable_hlo_dump_test_on_tpu -> origin/JackCao/Disable_hlo_dump_test_on_tpu
Step #0 - "git_fetch":  * [new branch]          JackCao/cherry_pick_dynamo_1 -> origin/JackCao/cherry_pick_dynamo_1
Step #0 - "git_fetch":  * [new branch]          JackCao/cherry_pick_dynamo_2 -> origin/JackCao/cherry_pick_dynamo_2
Step #0 - "git_fetch":  * [new branch]          JackCao/cherry_pick_dynamo_3 -> origin/JackCao/cherry_pick_dynamo_3
Step #0 - "git_fetch":  * [new branch]          JackCao/spmd_python_device -> origin/JackCao/spmd_python_device
Step #0 - "git_fetch":  * [new branch]          JackCao/unordered_sharding_spec -> origin/JackCao/unordered_sharding_spec
Step #0 - "git_fetch":  * [new branch]          JackCao/update_to_2.0   -> origin/JackCao/update_to_2.0
Step #0 - "git_fetch":  * [new branch]          JackCao/workaround_einsum_faketensor -> origin/JackCao/workaround_einsum_faketensor
Step #0 - "git_fetch":  * [new branch]          JackCaoG-patch-1        -> origin/JackCaoG-patch-1
Step #0 - "git_fetch":  * [new branch]          JackCaoG-patch-2        -> origin/JackCaoG-patch-2
Step #0 - "git_fetch":  * [new branch]          JackCaoG-patch-3        -> origin/JackCaoG-patch-3
Step #0 - "git_fetch":  * [new branch]          JackCaoG-patch-4        -> origin/JackCaoG-patch-4
Step #0 - "git_fetch":  * [new branch]          JackCaoG-patch-5        -> origin/JackCaoG-patch-5
Step #0 - "git_fetch":  * [new branch]          JackCaoG-remove-pin     -> origin/JackCaoG-remove-pin
Step #0 - "git_fetch":  * [new branch]          JackCaoG/1_13_revert_addcmul -> origin/JackCaoG/1_13_revert_addcmul
Step #0 - "git_fetch":  * [new branch]          JackCaoG/2.0_patch      -> origin/JackCaoG/2.0_patch
Step #0 - "git_fetch":  * [new branch]          JackCaoG/AddDynamoSpmdMetrics -> origin/JackCaoG/AddDynamoSpmdMetrics
Step #0 - "git_fetch":  * [new branch]          JackCaoG/CI_build_cpp_test -> origin/JackCaoG/CI_build_cpp_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/Dyanmo_execute_async -> origin/JackCaoG/Dyanmo_execute_async
Step #0 - "git_fetch":  * [new branch]          JackCaoG/Dynamo_training_test -> origin/JackCaoG/Dynamo_training_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/PJRT_wait_buffer_sync -> origin/JackCaoG/PJRT_wait_buffer_sync
Step #0 - "git_fetch":  * [new branch]          JackCaoG/SPMD_execute_metrics -> origin/JackCaoG/SPMD_execute_metrics
Step #0 - "git_fetch":  * [new branch]          JackCaoG/SPMD_int_to_cpu -> origin/JackCaoG/SPMD_int_to_cpu
Step #0 - "git_fetch":  * [new branch]          JackCaoG/Update_short_metrics -> origin/JackCaoG/Update_short_metrics
Step #0 - "git_fetch":  * [new branch]          JackCaoG/Update_wheel_docker_to_table -> origin/JackCaoG/Update_wheel_docker_to_table
Step #0 - "git_fetch":  * [new branch]          JackCaoG/add_2_1_py311_build -> origin/JackCaoG/add_2_1_py311_build
Step #0 - "git_fetch":  * [new branch]          JackCaoG/add_dynamo_test_counter_check -> origin/JackCaoG/add_dynamo_test_counter_check
Step #0 - "git_fetch":  * [new branch]          JackCaoG/add_dynamo_test_to_tpu_ci -> origin/JackCaoG/add_dynamo_test_to_tpu_ci
Step #0 - "git_fetch":  * [new branch]          JackCaoG/add_gpu_doc    -> origin/JackCaoG/add_gpu_doc
Step #0 - "git_fetch":  * [new branch]          JackCaoG/add_gpu_doc_readme -> origin/JackCaoG/add_gpu_doc_readme
Step #0 - "git_fetch":  * [new branch]          JackCaoG/add_test_dynamo_graph_dump -> origin/JackCaoG/add_test_dynamo_graph_dump
Step #0 - "git_fetch":  * [new branch]          JackCaoG/add_test_print -> origin/JackCaoG/add_test_print
Step #0 - "git_fetch":  * [new branch]          JackCaoG/api_init_runtime -> origin/JackCaoG/api_init_runtime
Step #0 - "git_fetch":  * [new branch]          JackCaoG/barrier_comment -> origin/JackCaoG/barrier_comment
Step #0 - "git_fetch":  * [new branch]          JackCaoG/better_dynamo_error_message -> origin/JackCaoG/better_dynamo_error_message
Step #0 - "git_fetch":  * [new branch]          JackCaoG/build_audio_whl -> origin/JackCaoG/build_audio_whl
Step #0 - "git_fetch":  * [new branch]          JackCaoG/bump+gpu_coverage_timeout -> origin/JackCaoG/bump+gpu_coverage_timeout
Step #0 - "git_fetch":  * [new branch]          JackCaoG/bump_build_timeout -> origin/JackCaoG/bump_build_timeout
Step #0 - "git_fetch":  * [new branch]          JackCaoG/cache_sharded_placeholder_dynamo -> origin/JackCaoG/cache_sharded_placeholder_dynamo
Step #0 - "git_fetch":  * [new branch]          JackCaoG/cherry_pick_dynamo_test -> origin/JackCaoG/cherry_pick_dynamo_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/cherry_picl_xrt_crash_2.1 -> origin/JackCaoG/cherry_picl_xrt_crash_2.1
Step #0 - "git_fetch":  * [new branch]          JackCaoG/clean_up_gettensor -> origin/JackCaoG/clean_up_gettensor
Step #0 - "git_fetch":  * [new branch]          JackCaoG/clean_up_use_virtual_device -> origin/JackCaoG/clean_up_use_virtual_device
Step #0 - "git_fetch":  * [new branch]          JackCaoG/cleanup_docs   -> origin/JackCaoG/cleanup_docs
Step #0 - "git_fetch":  * [new branch]          JackCaoG/colab_whl_abi  -> origin/JackCaoG/colab_whl_abi
Step #0 - "git_fetch":  * [new branch]          JackCaoG/colab_whl_abi_2 -> origin/JackCaoG/colab_whl_abi_2
Step #0 - "git_fetch":  * [new branch]          JackCaoG/data_to_string -> origin/JackCaoG/data_to_string
Step #0 - "git_fetch":  * [new branch]          JackCaoG/debug_tool     -> origin/JackCaoG/debug_tool
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_coverage -> origin/JackCaoG/disable_coverage
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_dynamo_different_shape_test -> origin/JackCaoG/disable_dynamo_different_shape_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_flaky_test -> origin/JackCaoG/disable_flaky_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_flaky_test_2 -> origin/JackCaoG/disable_flaky_test_2
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_flaky_test_3 -> origin/JackCaoG/disable_flaky_test_3
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_more_dynamic -> origin/JackCaoG/disable_more_dynamic
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_more_dynamic_test -> origin/JackCaoG/disable_more_dynamic_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disable_test_xla_sharding -> origin/JackCaoG/disable_test_xla_sharding
Step #0 - "git_fetch":  * [new branch]          JackCaoG/disbale_docker_sandbox -> origin/JackCaoG/disbale_docker_sandbox
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dump_output_sharding -> origin/JackCaoG/dump_output_sharding
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_2.0_cherry_pick -> origin/JackCaoG/dynamo_2.0_cherry_pick
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_debug_message -> origin/JackCaoG/dynamo_debug_message
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_different_graph -> origin/JackCaoG/dynamo_different_graph
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_doc     -> origin/JackCaoG/dynamo_doc
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_dump_tensor_file -> origin/JackCaoG/dynamo_dump_tensor_file
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_note_backport_2.1 -> origin/JackCaoG/dynamo_note_backport_2.1
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_only_warm_up_cache -> origin/JackCaoG/dynamo_only_warm_up_cache
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_optimizer -> origin/JackCaoG/dynamo_optimizer
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_reduce_execute -> origin/JackCaoG/dynamo_reduce_execute
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_rng_seed_without_mark_step -> origin/JackCaoG/dynamo_rng_seed_without_mark_step
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_spmd    -> origin/JackCaoG/dynamo_spmd
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_spmd_input_sharding_change -> origin/JackCaoG/dynamo_spmd_input_sharding_change
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_spmd_optimization -> origin/JackCaoG/dynamo_spmd_optimization
Step #0 - "git_fetch":  * [new branch]          JackCaoG/dynamo_warning -> origin/JackCaoG/dynamo_warning
Step #0 - "git_fetch":  * [new branch]          JackCaoG/enable_dynamo_upstream_test -> origin/JackCaoG/enable_dynamo_upstream_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/enable_fsdp_gpu_test -> origin/JackCaoG/enable_fsdp_gpu_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_cc_linter_once -> origin/JackCaoG/fix_cc_linter_once
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_deprecated_test_oepration_api -> origin/JackCaoG/fix_deprecated_test_oepration_api
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_dyanmo_additional_tracing -> origin/JackCaoG/fix_dyanmo_additional_tracing
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_dynamo     -> origin/JackCaoG/fix_dynamo
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_dynamo_crash -> origin/JackCaoG/fix_dynamo_crash
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_dynamo_inplace -> origin/JackCaoG/fix_dynamo_inplace
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_dynamo_text_allclose -> origin/JackCaoG/fix_dynamo_text_allclose
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_hlo_dump   -> origin/JackCaoG/fix_hlo_dump
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_hlo_dump_spmd -> origin/JackCaoG/fix_hlo_dump_spmd
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_mark_step_on_spmd_device -> origin/JackCaoG/fix_mark_step_on_spmd_device
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_mp_save_test -> origin/JackCaoG/fix_mp_save_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_send_cpu_data -> origin/JackCaoG/fix_send_cpu_data
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_spmd_buff_is_null -> origin/JackCaoG/fix_spmd_buff_is_null
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_tpu_ci_accuracy -> origin/JackCaoG/fix_tpu_ci_accuracy
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fix_vision_build -> origin/JackCaoG/fix_vision_build
Step #0 - "git_fetch":  * [new branch]          JackCaoG/fsdp_autowrap_test -> origin/JackCaoG/fsdp_autowrap_test
Step #0 - "git_fetch":  * [new branch]          JackCaoG/gpu_build_pytorch_cuda -> origin/JackCaoG/gpu_build_pytorch_cuda
Step #0 - "git_fetch":  * [new branch]          JackCaoG/handle_dynamo_fn_without_arg -> origin/JackCaoG/handle_dynamo_fn_without_arg
Step #0 - "git_fetch":  * [new branch]          JackCaoG/improve_data_tostring -> origin/JackCaoG/improve_data_tostring
Step #0 - "git_fetch":  * [new branch]          JackCaoG/increase_timeout_docker -> origin/JackCaoG/increase_timeout_docker
Step #0 - "git_fetch":  * [new branch]          JackCaoG/ir_value_sharding -> origin/JackCaoG/ir_value_sharding
Step #0 - "git_fetch":  * [new branch]          JackCaoG/lower_dropout_backward -> origin/JackCaoG/lower_dropout_backward
Step #0 - "git_fetch":  * [new branch]          JackCaoG/move_op_lowering_guide -> origin/JackCaoG/move_op_lowering_guide
Step #0 - "git_fetch":  * [new branch]          JackCaoG/move_pin_2_0   -> origin/JackCaoG/move_pin_2_0
Step #0 - "git_fetch":  * [new branch]          JackCaoG/move_pjrt_lock_location -> origin/JackCaoG/move_pjrt_lock_location
Step #0 - "git_fetch":  * [new branch]          JackCaoG/move_return_future -> origin/JackCaoG/move_return_future
Step #0 - "git_fetch":  * [new branch]          JackCaoG/new_alias_api  -> origin/JackCaoG/new_alias_api
Step #0 - "git_fetch":  * [new branch]          JackCaoG/nightly_build_python -> origin/JackCaoG/nightly_build_python
Step #0 - "git_fetch":  * [new branch]          JackCaoG/parallel_spmd  -> origin/JackCaoG/parallel_spmd
Step #0 - "git_fetch":  * [new branch]          JackCaoG/pin_update_20230808 -> origin/JackCaoG/pin_update_20230808
Step #0 - "git_fetch":  * [new branch]          JackCaoG/pt_xla_debug_5 -> origin/JackCaoG/pt_xla_debug_5
Step #0 - "git_fetch":  * [new branch]          JackCaoG/pt_xla_debug_dynamo_execute -> origin/JackCaoG/pt_xla_debug_dynamo_execute
Step #0 - "git_fetch":  * [new branch]          JackCaoG/r20_backportdoc -> origin/JackCaoG/r20_backportdoc
Step #0 - "git_fetch":  * [new branch]          JackCaoG/r2_0_ABI_0     -> origin/JackCaoG/r2_0_ABI_0
Step #0 - "git_fetch":  * [new branch]          JackCaoG/r2_0_back_port_profile_flag -> origin/JackCaoG/r2_0_back_port_profile_flag
Step #0 - "git_fetch":  * [new branch]          JackCaoG/r2_0_backport_loader_thread -> origin/JackCaoG/r2_0_backport_loader_thread
Step #0 - "git_fetch":  * [new branch]          JackCaoG/r2_0_backport_loader_thread_2 -> origin/JackCaoG/r2_0_backport_loader_thread_2
Step #0 - "git_fetch":  * [new branch]          JackCaoG/r2_0_cherry_pick_defer_init -> origin/JackCaoG/r2_0_cherry_pick_defer_init
Step #0 - "git_fetch":  * [new branch]          JackCaoG/remove_pin     -> origin/JackCaoG/remove_pin
Step #0 - "git_fetch":  * [new branch]          JackCaoG/restrict_mark_sharding -> origin/JackCaoG/restrict_mark_sharding
Step #0 - "git_fetch":  * [new branch]          JackCaoG/revert_addcmul -> origin/JackCaoG/revert_addcmul
Step #0 - "git_fetch":  * [new branch]          JackCaoG/revert_protobuf_tf_patch -> origin/JackCaoG/revert_protobuf_tf_patch
Step #0 - "git_fetch":  * [new branch]          JackCaoG/shard_device_data_ir -> origin/JackCaoG/shard_device_data_ir
Step #0 - "git_fetch":  * [new branch]          JackCaoG/skip_partitioner -> origin/JackCaoG/skip_partitioner
Step #0 - "git_fetch":  * [new branch]          JackCaoG/skip_tpu_test_precision -> origin/JackCaoG/skip_tpu_test_precision
Step #0 - "git_fetch":  * [new branch]          JackCaoG/spmd_aliasing  -> origin/JackCaoG/spmd_aliasing
Step #0 - "git_fetch":  * [new branch]          JackCaoG/spmd_dynamo_sharding_changed -> origin/JackCaoG/spmd_dynamo_sharding_changed
Step #0 - "git_fetch":  * [new branch]          JackCaoG/spmd_incorrect_result -> origin/JackCaoG/spmd_incorrect_result
Step #0 - "git_fetch":  * [new branch]          JackCaoG/tensor_debugging_api -> origin/JackCaoG/tensor_debugging_api
Step #0 - "git_fetch":  * [new branch]          JackCaoG/tensor_with_xla_data_sharding -> origin/JackCaoG/tensor_with_xla_data_sharding
Step #0 - "git_fetch":  * [new branch]          JackCaoG/test_torch_compile_usage -> origin/JackCaoG/test_torch_compile_usage
Step #0 - "git_fetch":  * [new branch]          JackCaoG/tf_update_20230201 -> origin/JackCaoG/tf_update_20230201
Step #0 - "git_fetch":  * [new branch]          JackCaoG/tracing_timer  -> origin/JackCaoG/tracing_timer
Step #0 - "git_fetch":  * [new branch]          JackCaoG/truncate_python_stack -> origin/JackCaoG/truncate_python_stack
Step #0 - "git_fetch":  * [new branch]          JackCaoG/unordered_partial_sharding_spec -> origin/JackCaoG/unordered_partial_sharding_spec
Step #0 - "git_fetch":  * [new branch]          JackCaoG/updateDocker   -> origin/JackCaoG/updateDocker
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_2_0_ABI -> origin/JackCaoG/update_2_0_ABI
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_API_GUIDE -> origin/JackCaoG/update_API_GUIDE
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_XLA_IR_SHAPE_CACHE_SIZE -> origin/JackCaoG/update_XLA_IR_SHAPE_CACHE_SIZE
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_build_guide -> origin/JackCaoG/update_build_guide
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_colab_2_10_whl -> origin/JackCaoG/update_colab_2_10_whl
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_doc1    -> origin/JackCaoG/update_doc1
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_doc_remove_xrt -> origin/JackCaoG/update_doc_remove_xrt
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_docker_instruction -> origin/JackCaoG/update_docker_instruction
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_dynamo_backend_name -> origin/JackCaoG/update_dynamo_backend_name
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_dynamo_backend_nograd -> origin/JackCaoG/update_dynamo_backend_nograd
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_dynamo_doc -> origin/JackCaoG/update_dynamo_doc
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_dynamo_number_2.1 -> origin/JackCaoG/update_dynamo_number_2.1
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_dynamo_readme -> origin/JackCaoG/update_dynamo_readme
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_dynamo_save_graph_path -> origin/JackCaoG/update_dynamo_save_graph_path
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_dynamo_torch_equal -> origin/JackCaoG/update_dynamo_torch_equal
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_gpu_doc -> origin/JackCaoG/update_gpu_doc
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_gpu_wheel_link -> origin/JackCaoG/update_gpu_wheel_link
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_gpu_wheellink -> origin/JackCaoG/update_gpu_wheellink
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_pjrt_future -> origin/JackCaoG/update_pjrt_future
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_readme_2 -> origin/JackCaoG/update_readme_2
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_set_default_tensor_type -> origin/JackCaoG/update_set_default_tensor_type
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_tf_2023018 -> origin/JackCaoG/update_tf_2023018
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_to_2.2  -> origin/JackCaoG/update_to_2.2
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_torch_compile_usage -> origin/JackCaoG/update_torch_compile_usage
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_trouble_shotting_doc -> origin/JackCaoG/update_trouble_shotting_doc
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_troubleshotting_with_sanitycheck -> origin/JackCaoG/update_troubleshotting_with_sanitycheck
Step #0 - "git_fetch":  * [new branch]          JackCaoG/update_wheel_docker_format -> origin/JackCaoG/update_wheel_docker_format
Step #0 - "git_fetch":  * [new branch]          JackCaoG/use_new_alias_api -> origin/JackCaoG/use_new_alias_api
Step #0 - "git_fetch":  * [new branch]          JackCaoG/xrt_cherry_pick_07_10 -> origin/JackCaoG/xrt_cherry_pick_07_10
Step #0 - "git_fetch":  * [new branch]          JackCaoG/xrt_cherrypick_0721 -> origin/JackCaoG/xrt_cherrypick_0721
Step #0 - "git_fetch":  * [new branch]          JackCaog/dynamo_input_sharding -> origin/JackCaog/dynamo_input_sharding
Step #0 - "git_fetch":  * [new branch]          JackCaog/quick_spmd_fix -> origin/JackCaog/quick_spmd_fix
Step #0 - "git_fetch":  * [new branch]          JackCaog/spmd_device_default -> origin/JackCaog/spmd_device_default
Step #0 - "git_fetch":  * [new branch]          Liyang90-patch-1        -> origin/Liyang90-patch-1
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-1       -> origin/ManfeiBai-patch-1
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-10      -> origin/ManfeiBai-patch-10
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-11      -> origin/ManfeiBai-patch-11
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-12      -> origin/ManfeiBai-patch-12
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-13      -> origin/ManfeiBai-patch-13
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-14      -> origin/ManfeiBai-patch-14
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-15      -> origin/ManfeiBai-patch-15
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-16      -> origin/ManfeiBai-patch-16
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-17      -> origin/ManfeiBai-patch-17
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-18      -> origin/ManfeiBai-patch-18
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-19      -> origin/ManfeiBai-patch-19
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-2       -> origin/ManfeiBai-patch-2
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-20      -> origin/ManfeiBai-patch-20
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-21      -> origin/ManfeiBai-patch-21
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-22      -> origin/ManfeiBai-patch-22
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-23      -> origin/ManfeiBai-patch-23
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-24      -> origin/ManfeiBai-patch-24
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-25      -> origin/ManfeiBai-patch-25
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-26      -> origin/ManfeiBai-patch-26
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-27      -> origin/ManfeiBai-patch-27
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-28      -> origin/ManfeiBai-patch-28
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-29      -> origin/ManfeiBai-patch-29
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-3       -> origin/ManfeiBai-patch-3
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-30      -> origin/ManfeiBai-patch-30
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-31      -> origin/ManfeiBai-patch-31
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-32      -> origin/ManfeiBai-patch-32
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-33      -> origin/ManfeiBai-patch-33
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-34      -> origin/ManfeiBai-patch-34
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-35      -> origin/ManfeiBai-patch-35
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-36      -> origin/ManfeiBai-patch-36
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-37      -> origin/ManfeiBai-patch-37
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-38      -> origin/ManfeiBai-patch-38
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-39      -> origin/ManfeiBai-patch-39
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-4       -> origin/ManfeiBai-patch-4
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-40      -> origin/ManfeiBai-patch-40
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-41      -> origin/ManfeiBai-patch-41
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-5       -> origin/ManfeiBai-patch-5
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-6       -> origin/ManfeiBai-patch-6
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-7       -> origin/ManfeiBai-patch-7
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-8       -> origin/ManfeiBai-patch-8
Step #0 - "git_fetch":  * [new branch]          ManfeiBai-patch-9       -> origin/ManfeiBai-patch-9
Step #0 - "git_fetch":  * [new branch]          XLASymNodeImplSub       -> origin/XLASymNodeImplSub
Step #0 - "git_fetch":  * [new branch]          XLA_HLO_DEBUG_METADATA  -> origin/XLA_HLO_DEBUG_METADATA
Step #0 - "git_fetch":  * [new branch]          a_tensor_cleanup        -> origin/a_tensor_cleanup
Step #0 - "git_fetch":  * [new branch]          accu_rate               -> origin/accu_rate
Step #0 - "git_fetch":  * [new branch]          adapt_lazy_fun_1        -> origin/adapt_lazy_fun_1
Step #0 - "git_fetch":  * [new branch]          add-imagenet-learning-rate-optimizer -> origin/add-imagenet-learning-rate-optimizer
Step #0 - "git_fetch":  * [new branch]          add-imagenet-learning-rate-scheduler -> origin/add-imagenet-learning-rate-scheduler
Step #0 - "git_fetch":  * [new branch]          add-imagenet-learning-schedule -> origin/add-imagenet-learning-schedule
Step #0 - "git_fetch":  * [new branch]          add-summary-writer      -> origin/add-summary-writer
Step #0 - "git_fetch":  * [new branch]          addAddDivToXLASymIntNodeImpl -> origin/addAddDivToXLASymIntNodeImpl
Step #0 - "git_fetch":  * [new branch]          addAnotherExpandSymintPyTest -> origin/addAnotherExpandSymintPyTest
Step #0 - "git_fetch":  * [new branch]          addAnsibleToR112        -> origin/addAnsibleToR112
Step #0 - "git_fetch":  * [new branch]          addAnsibleToR113        -> origin/addAnsibleToR113
Step #0 - "git_fetch":  * [new branch]          addDSModelTest          -> origin/addDSModelTest
Step #0 - "git_fetch":  * [new branch]          addExpandSymintTest     -> origin/addExpandSymintTest
Step #0 - "git_fetch":  * [new branch]          addMoreTestsForSizeAdd  -> origin/addMoreTestsForSizeAdd
Step #0 - "git_fetch":  * [new branch]          addPatch                -> origin/addPatch
Step #0 - "git_fetch":  * [new branch]          addPin                  -> origin/addPin
Step #0 - "git_fetch":  * [new branch]          addPy310Cuda118NightlyWheel -> origin/addPy310Cuda118NightlyWheel
Step #0 - "git_fetch":  * [new branch]          addTestForNonzero       -> origin/addTestForNonzero
Step #0 - "git_fetch":  * [new branch]          addVersionSelectionToR111 -> origin/addVersionSelectionToR111
Step #0 - "git_fetch":  * [new branch]          addVersionSelectionToR112 -> origin/addVersionSelectionToR112
Step #0 - "git_fetch":  * [new branch]          addVersionSelectionToR113 -> origin/addVersionSelectionToR113
Step #0 - "git_fetch":  * [new branch]          add_gitignore           -> origin/add_gitignore
Step #0 - "git_fetch":  * [new branch]          add_gitrev              -> origin/add_gitrev
Step #0 - "git_fetch":  * [new branch]          add_ne_size_node        -> origin/add_ne_size_node
Step #0 - "git_fetch":  * [new branch]          add_os_import_cifar     -> origin/add_os_import_cifar
Step #0 - "git_fetch":  * [new branch]          add_py_bench            -> origin/add_py_bench
Step #0 - "git_fetch":  * [new branch]          add_rendezvous_test     -> origin/add_rendezvous_test
Step #0 - "git_fetch":  * [new branch]          add_roots_tags          -> origin/add_roots_tags
Step #0 - "git_fetch":  * [new branch]          add_set_dimension_sizes -> origin/add_set_dimension_sizes
Step #0 - "git_fetch":  * [new branch]          add_symint_to_ci_docker -> origin/add_symint_to_ci_docker
Step #0 - "git_fetch":  * [new branch]          add_symint_util         -> origin/add_symint_util
Step #0 - "git_fetch":  * [new branch]          add_telemetry_env_var   -> origin/add_telemetry_env_var
Step #0 - "git_fetch":  * [new branch]          add_test_filter         -> origin/add_test_filter
Step #0 - "git_fetch":  * [new branch]          add_toxla_test          -> origin/add_toxla_test
Step #0 - "git_fetch":  * [new branch]          add_zach_comment        -> origin/add_zach_comment
Step #0 - "git_fetch":  * [new branch]          addcmul                 -> origin/addcmul
Step #0 - "git_fetch":  * [new branch]          addneg                  -> origin/addneg
Step #0 - "git_fetch":  * [new branch]          addr                    -> origin/addr
Step #0 - "git_fetch":  * [new branch]          aifix_warnings          -> origin/aifix_warnings
Step #0 - "git_fetch":  * [new branch]          ailing_fix_sums         -> origin/ailing_fix_sums
Step #0 - "git_fetch":  * [new branch]          ailzhang-patch-1        -> origin/ailzhang-patch-1
Step #0 - "git_fetch":  * [new branch]          alanwaketan/addcdiv     -> origin/alanwaketan/addcdiv
Step #0 - "git_fetch":  * [new branch]          alanwaketan/alias       -> origin/alanwaketan/alias
Step #0 - "git_fetch":  * [new branch]          alanwaketan/all_gather  -> origin/alanwaketan/all_gather
Step #0 - "git_fetch":  * [new branch]          alanwaketan/all_gather_token -> origin/alanwaketan/all_gather_token
Step #0 - "git_fetch":  * [new branch]          alanwaketan/all_reduce  -> origin/alanwaketan/all_reduce
Step #0 - "git_fetch":  * [new branch]          alanwaketan/all_reduce_d -> origin/alanwaketan/all_reduce_d
Step #0 - "git_fetch":  * [new branch]          alanwaketan/all_reduce_t -> origin/alanwaketan/all_reduce_t
Step #0 - "git_fetch":  * [new branch]          alanwaketan/as_strided  -> origin/alanwaketan/as_strided
Step #0 - "git_fetch":  * [new branch]          alanwaketan/as_strided_2 -> origin/alanwaketan/as_strided_2
Step #0 - "git_fetch":  * [new branch]          alanwaketan/build_input_output -> origin/alanwaketan/build_input_output
Step #0 - "git_fetch":  * [new branch]          alanwaketan/cc          -> origin/alanwaketan/cc
Step #0 - "git_fetch":  * [new branch]          alanwaketan/ci          -> origin/alanwaketan/ci
Step #0 - "git_fetch":  * [new branch]          alanwaketan/conv        -> origin/alanwaketan/conv
Step #0 - "git_fetch":  * [new branch]          alanwaketan/copy        -> origin/alanwaketan/copy
Step #0 - "git_fetch":  * [new branch]          alanwaketan/cpp_test    -> origin/alanwaketan/cpp_test

...
[Logs truncated due to log size limitations. For full logs, see https://console.cloud.google.com/cloud-build/builds;region=us-central1/ca3303d1-656b-40dc-ab0b-d752147b147f?project=164006649440.]
...
0 step 4080 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4090 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4100 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4110 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4120 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4130 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4140 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4150 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4160 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4170 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4180 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4190 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4200 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4210 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4220 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4230 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4240 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4250 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4260 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4270 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4280 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4290 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4300 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4310 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4320 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4330 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4340 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4350 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4360 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4370 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4380 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4390 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4400 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4410 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4420 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4430 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4440 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4450 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4460 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4470 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4480 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4490 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4500 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4510 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4520 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4530 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4540 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4550 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4560 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4570 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4580 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4590 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4600 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4610 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4620 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4630 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4640 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4650 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4660 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4670 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4680 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4690 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4700 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4710 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4720 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4730 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4740 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4750 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4760 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4770 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4780 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4790 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4800 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4810 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4820 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4830 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4840 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4850 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4860 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4870 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4880 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4890 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4900 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4910 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4920 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4930 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4940 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4950 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4960 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4970 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4980 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 4990 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5000 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5010 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5020 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5030 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5040 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5050 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5060 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5070 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5080 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5090 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5100 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5110 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5120 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5130 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5140 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5150 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5160 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5170 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5180 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5190 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5200 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5210 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5220 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5230 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5240 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5250 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5260 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5270 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5280 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5290 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5300 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5310 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5320 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5330 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5340 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5350 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5360 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5370 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5380 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5390 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5400 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5410 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5420 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5430 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5440 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5450 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5460 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5470 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5480 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5490 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5500 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5510 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5520 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5530 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5540 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5550 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5560 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5570 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5580 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5590 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5600 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5610 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5620 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5630 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5640 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5650 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5660 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5670 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5680 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5690 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5700 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5710 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5720 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5730 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5740 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5750 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5760 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5770 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5780 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5790 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5800 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5810 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5820 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5830 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5840 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5850 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5860 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5870 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5880 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5890 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5900 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5910 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5920 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5930 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5940 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5950 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5960 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5970 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5980 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 5990 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6000 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6010 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6020 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6030 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6040 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6050 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6060 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6070 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6080 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6090 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6100 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6110 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6120 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6130 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6140 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6150 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6160 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6170 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6180 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6190 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6200 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6210 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6220 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6230 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6240 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6250 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6260 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6270 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6280 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6290 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6300 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6310 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6320 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6330 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6340 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6350 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6360 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6370 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6380 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6390 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6400 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6410 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6420 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6430 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6440 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6450 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6460 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6470 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6480 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6490 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6500 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6510 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6520 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6530 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6540 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6550 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6560 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6570 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6580 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6590 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6600 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6610 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6620 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6630 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6640 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6650 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6660 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6670 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6680 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6690 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6700 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6710 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6720 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6730 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6740 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6750 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6760 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6770 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6780 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6790 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6800 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6810 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6820 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6830 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6840 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6850 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6860 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6870 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6880 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6890 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6900 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6910 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6920 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6930 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6940 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6950 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6960 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6970 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6980 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 6990 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7000 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7010 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7020 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7030 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7040 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7050 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7060 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7070 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7080 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7090 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7100 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7110 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7120 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7130 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7140 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7150 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7160 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7170 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7180 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7190 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7200 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7210 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7220 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7230 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7240 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7250 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7260 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7270 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7280 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7290 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7300 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7310 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7320 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7330 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7340 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7350 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7360 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7370 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7380 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7390 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7400 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7410 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7420 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7430 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7440 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7450 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7460 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7470 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7480 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7490 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7500 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7510 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7520 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7530 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7540 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7550 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7560 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7570 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7580 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7590 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7600 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7610 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7620 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7630 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7640 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7650 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7660 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7670 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7680 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7690 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7700 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7710 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7720 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7730 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7740 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7750 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7760 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7770 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7780 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7790 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7800 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7810 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7820 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7830 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7840 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7850 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7860 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7870 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7880 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7890 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7900 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7910 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7920 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7930 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7940 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7950 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7960 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7970 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7980 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 7990 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8000 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8010 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8020 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8030 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8040 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8050 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8060 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8070 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8080 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8090 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8100 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8110 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8120 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8130 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8140 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8150 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8160 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8170 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8180 loss 0.0
Step #4 - "run_e2e_tests": Epoch 0 step 8190 loss 0.0
Step #4 - "run_e2e_tests": + python3 /src/pytorch/xla/test/spmd/test_xla_spmd_python_api_interaction.py
Step #4 - "run_e2e_tests": ./usr/local/lib/python3.10/site-packages/torch_xla/runtime.py:233: UserWarning: XLA_USE_SPMD is being deprecated. Use torch_xla.runtime.use_spmd() without setting XLA_USE_SPMD env-var.
Step #4 - "run_e2e_tests":   warnings.warn("XLA_USE_SPMD is being deprecated. "
Step #4 - "run_e2e_tests": ......................
Step #4 - "run_e2e_tests": ----------------------------------------------------------------------
Step #4 - "run_e2e_tests": Ran 23 tests in 5.849s
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": OK
Step #4 - "run_e2e_tests": + XLA_EXPERIMENTAL=nonzero:masked_select
Step #4 - "run_e2e_tests": + python3 /src/pytorch/xla/test/ds/test_dynamic_shape_models.py -v
Step #4 - "run_e2e_tests": test_backward_pass_with_dynamic_input (__main__.TestDynamicShapeModels) ... ok
Step #4 - "run_e2e_tests": test_forward_pass_dynamic_input_compile_once (__main__.TestDynamicShapeModels) ... ok
Step #4 - "run_e2e_tests": test_forward_pass_dynamic_input_correctness (__main__.TestDynamicShapeModels) ... ok
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": ----------------------------------------------------------------------
Step #4 - "run_e2e_tests": Ran 3 tests in 1.542s
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": OK
Step #4 - "run_e2e_tests": Finished training.
Step #4 - "run_e2e_tests": Test passed.
Step #4 - "run_e2e_tests": Test passed.
Step #4 - "run_e2e_tests": + XLA_EXPERIMENTAL=nonzero:masked_select
Step #4 - "run_e2e_tests": + python3 /src/pytorch/xla/test/ds/test_dynamic_shapes.py -v
Step #4 - "run_e2e_tests": test_SizeEq_should_not_compile_for_identical_symints (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_abs (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_add_dyn_with_static_broadcastable (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_add_dyn_with_static_not_broadcastable (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_add_two_dynamic_tensors (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_clone (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_empty_symint (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_expand_symint_correctness (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_fill_ (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_masked_select_shape (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_nonzero_cast (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_nonzero_correctness (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_nonzero_shape (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_simple_expand (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_simple_expand_add_dimension (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_simple_expand_on_2d_tensor (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sizeAdd (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sizeGe (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sizeLt (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sizeMod (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sizeNe (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sizeSub (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sub_dyn_with_static_broadcastable (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sub_dyn_with_static_not_broadcastable (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_sub_two_dynamic_tensors (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_t_copy (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_unsqueeze_copy_dynamism (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_view_copy_symint_with_dyn_input_dyn_input_shape (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_view_copy_symint_with_dyn_input_static_input_shape (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_view_copy_symint_with_static_input_dyn_input_shape (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_view_copy_symint_with_static_input_dyn_input_shape2 (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_wrap (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": test_xlatensor_memoize_symsizes (__main__.TestDynamicShapes) ... ok
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": ----------------------------------------------------------------------
Step #4 - "run_e2e_tests": Ran 33 tests in 1.717s
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": OK
Step #4 - "run_e2e_tests": + python3 /src/pytorch/xla/test/test_autocast.py
Step #4 - "run_e2e_tests": test_autocast_banned (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_linalg_fp16 (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_methods_expect_builtin_promote (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_methods_fp16 (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_methods_fp32 (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_nn_fp16 (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_nn_fp32 (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_torch_bf16 (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_torch_expect_builtin_promote (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_torch_fp32 (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_torch_need_autocast_promote (__main__.TestAutocastCuda) ... skipped 'CUDA autocast test.'
Step #4 - "run_e2e_tests": test_autocast_methods_bf16 (__main__.TestAutocastTPU) ... ok
Step #4 - "run_e2e_tests": test_autocast_methods_expect_builtin_promote (__main__.TestAutocastTPU) ... ok
Step #4 - "run_e2e_tests": test_autocast_methods_fp32 (__main__.TestAutocastTPU) ... ok
Step #4 - "run_e2e_tests": test_autocast_nn_fp32 (__main__.TestAutocastTPU) ... ok
Step #4 - "run_e2e_tests": test_autocast_torch_expect_builtin_promote (__main__.TestAutocastTPU) ... ok
Step #4 - "run_e2e_tests": test_autocast_torch_fp32 (__main__.TestAutocastTPU) ... ok
Step #4 - "run_e2e_tests": test_autocast_torch_need_autocast_promote (__main__.TestAutocastTPU) ... ok
Step #4 - "run_e2e_tests": test_autocast_tpu_check_dtype (__main__.TestAutocastTPU) ... /usr/local/lib/python3.10/site-packages/torch_xla/amp/autocast_mode.py:57: UserWarning: In XLA:TPU autocast, but the target dtype is not supported. Disabling autocast.
Step #4 - "run_e2e_tests": TPU Autocast only supports dtype of torch.bfloat16 currently.
Step #4 - "run_e2e_tests":   warnings.warn(error_message)
Step #4 - "run_e2e_tests": ok
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": ----------------------------------------------------------------------
Step #4 - "run_e2e_tests": Ran 19 tests in 79.214s
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": OK (skipped=11)
Step #4 - "run_e2e_tests": Fail to import hypothesis in common_utils, tests are not derandomized
Step #4 - "run_e2e_tests": + python3 /src/pytorch/xla/test/dynamo/test_dynamo.py
Step #4 - "run_e2e_tests": /usr/local/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: 'libc10_cuda.so: cannot open shared object file: No such file or directory'If you don't plan on using image functionality from `torchvision.io`, you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have `libjpeg` or `libpng` installed before building `torchvision` from source?
Step #4 - "run_e2e_tests":   warn(
Step #4 - "run_e2e_tests": ........s...s...
Step #4 - "run_e2e_tests": ----------------------------------------------------------------------
Step #4 - "run_e2e_tests": Ran 16 tests in 39.016s
Step #4 - "run_e2e_tests": 
Step #4 - "run_e2e_tests": OK (skipped=2)
Step #4 - "run_e2e_tests": ++ kubectl get pod/xla-test-job-g5vgj -o 'jsonpath={.status.containerStatuses[?(@.name=="xla-test")].state.terminated.exitCode}'
Step #4 - "run_e2e_tests": + exit 0
Finished Step #4 - "run_e2e_tests"
PUSH
DONE

Build Log: https://console.cloud.google.com/cloud-build/builds;region=us-central1/ca3303d1-656b-40dc-ab0b-d752147b147f?project=164006649440