Skip to content

autotuner: iterate in deterministic order#36755

Closed
olupton wants to merge 1 commit into
openxla:mainfrom
olupton:autotuner-determinism
Closed

autotuner: iterate in deterministic order#36755
olupton wants to merge 1 commit into
openxla:mainfrom
olupton:autotuner-determinism

Conversation

@olupton

@olupton olupton commented Jan 23, 2026

Copy link
Copy Markdown
Contributor

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None

@shyamliagrawal shyamliagrawal self-assigned this Jan 23, 2026
@shyamliagrawal shyamliagrawal self-requested a review January 23, 2026 14:50
@shyamliagrawal shyamliagrawal removed their assignment Jan 23, 2026

@shyamliagrawal shyamliagrawal left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the fix!

copybara-service Bot pushed a commit that referenced this pull request Jan 23, 2026
Imported from GitHub PR #36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

FUTURE_COPYBARA_INTEGRATE_REVIEW=#36755 from olupton:autotuner-determinism 3fa358c
PiperOrigin-RevId: 860091136
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Jan 23, 2026
Imported from GitHub PR openxla/xla#36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c373aa262d69f0f7cd43676a60d65395c3 by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#36755 from olupton:autotuner-determinism 3fa358c373aa262d69f0f7cd43676a60d65395c3
PiperOrigin-RevId: 860091136
copybara-service Bot pushed a commit that referenced this pull request Jan 23, 2026
Imported from GitHub PR #36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

FUTURE_COPYBARA_INTEGRATE_REVIEW=#36755 from olupton:autotuner-determinism 3fa358c
PiperOrigin-RevId: 860091136
copybara-service Bot pushed a commit that referenced this pull request Jan 23, 2026
Imported from GitHub PR #36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

FUTURE_COPYBARA_INTEGRATE_REVIEW=#36755 from olupton:autotuner-determinism 3fa358c
PiperOrigin-RevId: 860091136
copybara-service Bot pushed a commit that referenced this pull request Jan 23, 2026
Imported from GitHub PR #36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

FUTURE_COPYBARA_INTEGRATE_REVIEW=#36755 from olupton:autotuner-determinism 3fa358c
PiperOrigin-RevId: 860091136
copybara-service Bot pushed a commit that referenced this pull request Jan 23, 2026
Imported from GitHub PR #36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

FUTURE_COPYBARA_INTEGRATE_REVIEW=#36755 from olupton:autotuner-determinism 3fa358c
PiperOrigin-RevId: 860091136
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Jan 23, 2026
Imported from GitHub PR openxla/xla#36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c373aa262d69f0f7cd43676a60d65395c3 by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#36755 from olupton:autotuner-determinism 3fa358c373aa262d69f0f7cd43676a60d65395c3
PiperOrigin-RevId: 860091136
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Jan 23, 2026
FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#36755 from olupton:autotuner-determinism 3fa358c373aa262d69f0f7cd43676a60d65395c3
PiperOrigin-RevId: 840138208
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Jan 23, 2026
Imported from GitHub PR openxla/xla#36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c373aa262d69f0f7cd43676a60d65395c3 by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes #36755

PiperOrigin-RevId: 860161547
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Jan 23, 2026
FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#36755 from olupton:autotuner-determinism 3fa358c373aa262d69f0f7cd43676a60d65395c3
PiperOrigin-RevId: 856437764
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Jan 23, 2026
…removing unused inputs in ConditionalSimplifier pass

This updates the original value of a conditional op when unused operands are removed.

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#36755 from olupton:autotuner-determinism 3fa358c373aa262d69f0f7cd43676a60d65395c3
PiperOrigin-RevId: 859796297
nurmukhametov pushed a commit to ROCm/xla that referenced this pull request Feb 9, 2026
Imported from GitHub PR openxla#36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes openxla#36755

COPYBARA_INTEGRATE_REVIEW=openxla#36755 from olupton:autotuner-determinism 3fa358c
PiperOrigin-RevId: 860161547
nurmukhametov pushed a commit to ROCm/xla that referenced this pull request Feb 10, 2026
Imported from GitHub PR openxla#36755

📝 Summary of Changes
Avoid iterating in a run-to-run-unstable order (https://abseil.io/docs/cpp/guides/container#iteration-order-instability). This was causing different processes in distributed JAX runs to use inconsistent operation names.

🎯 Justification
PGLE expects consistent operation names.

🚀 Kind of Contribution
🐛 Bug Fix

🧪 Unit Tests:
None

🧪 Execution Tests:
None
Copybara import of the project:

--
3fa358c by Olli Lupton <olupton@nvidia.com>:

autotuner: iterate in deterministic order

Merging this change closes openxla#36755

COPYBARA_INTEGRATE_REVIEW=openxla#36755 from olupton:autotuner-determinism 3fa358c
PiperOrigin-RevId: 860161547
sergachev added a commit that referenced this pull request Mar 16, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants