Skip to content
Back to pull request #24387

[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. #64223

Sign in to view logs

[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation.

[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. #64223

Triggered via pull request October 18, 2024 17:28
@copybara-servicecopybara-service[bot]
synchronize #24387
test_687174276
Status Success
Total duration 11m 40s
Artifacts –

ci-build.yaml

on: pull_request
lint_and_typecheck
1m 0s
lint_and_typecheck
FFI example
45s
FFI example
Matrix: build
Matrix: Documentation - test code snippets
Matrix: Documentation - render documentation
Matrix: jax2tf_test
Fit to window
Zoom out
Zoom in

Annotations

2 errors
build with 3.13 (py 3.13 on ubuntu-20.04, x64=0)
WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable.It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning.
build with 3.10 (py 3.10 on ubuntu-20.04, x64=1)
WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv