Skip to content

feat: pass in new options to optimize-comm #1172

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

Merged
merged 14 commits into from
Apr 13, 2025
Merged

feat: pass in new options to optimize-comm #1172

merged 14 commits into from
Apr 13, 2025

Conversation

avik-pal
Copy link
Collaborator

needs EnzymeAD/Enzyme-JAX#773 + new JLL

@avik-pal avik-pal force-pushed the ap/pad_comm_opts branch 2 times, most recently from 9f3edba to 6875d18 Compare April 13, 2025 13:54
@avik-pal avik-pal requested a review from wsmoses April 13, 2025 13:56
@avik-pal avik-pal marked this pull request as ready for review April 13, 2025 13:56
@wsmoses wsmoses merged commit a1d868b into main Apr 13, 2025
6 of 11 checks passed
@wsmoses wsmoses deleted the ap/pad_comm_opts branch April 13, 2025 23:41
@@ -1459,7 +1464,7 @@ function compile_mlir!(
sym_visibility=MLIR.IR.attr(compiled_f, "private"),
)
fnbody = MLIR.IR.Block(
in_tys_padded, [MLIR.IR.Location() for _ in in_tys_padded]
in_tys_padded, [MLIR.IR.Location(MLIR.API.mlirValueGetLocation(MLIR.IR.argument(MLIR.IR.first_block(MLIR.IR.region(compiled_f, 1)), i))) for i in 1:length(linear_args)]
Copy link
Contributor

Choose a reason for hiding this comment

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

[JuliaFormatter] reported by reviewdog 🐶

Suggested change
in_tys_padded, [MLIR.IR.Location(MLIR.API.mlirValueGetLocation(MLIR.IR.argument(MLIR.IR.first_block(MLIR.IR.region(compiled_f, 1)), i))) for i in 1:length(linear_args)]
in_tys_padded,
[
MLIR.IR.Location(
MLIR.API.mlirValueGetLocation(
MLIR.IR.argument(
MLIR.IR.first_block(MLIR.IR.region(compiled_f, 1)), i
),
),
) for i in 1:length(linear_args)
],

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