add pull_request event #4
Open
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This ensures all existing JAX buffer types have a `delete` method that can be used to free device buffer allocation eagerly. User code sometimes have lingering python refs due to cyclic deps and other reasons, yet users may know for sure that certain arrays will no longer be used after a certain point. Calling `foo_array.delete()` for DeviceArray/ShardedDeviceArray/GlobalDeviceArray/Array allows users to force free the device side allocation to minimize device memory usage. PiperOrigin-RevId: 482892157
…e in CUDA 11.1 PiperOrigin-RevId: 482897448
PiperOrigin-RevId: 482902569
PiperOrigin-RevId: 482903592
PiperOrigin-RevId: 482905407
…ther than trivial computation. PiperOrigin-RevId: 482919649
PiperOrigin-RevId: 482945880
The shape function of DotGeneralOp can't be integrated into MHLO yet: the shape function only predicts return shape but not able to predict element type. However, the current python binding infra will generate the constructor __init__() without the `return` as the first arg, which assumes the shape function can provide a fully inferred type (including an accurate element type). This leads to "inferred type does not match actual result type" errors in JAX. This needs a future solution. This CL is the corresponding change with openxla/stablehlo#269 Related Python __init__() interface changes (used by JAX): batch_norm_grad: not used by JAX batch_norm_inference: not used by JAX batch_norm_training: not used by JAX case: no change* dot_general: open new b/253644255 to track the issue if: no change* map: no change* reduce: no change* reduce_window: no change* sort: no change* triangular_solve: updated in `linalg.py` while: no change* no change*: the signature of __init()__ for the op is not changed because of existence of regions https://github.com/llvm/llvm-project/blob/main/mlir/tools/mlir-tblgen/OpPythonBindingGen.cpp#L577 PiperOrigin-RevId: 482951512
fix some shape and type issues import into namespace imports into non-_src library working logpdf test cleanup working tests for cdf and sf after fixing select relax need for x to be in (a, b) ensure behavior with invalid input matches scipy remove enforcing valid parameters in tests added truncnorm to docs whoops alphabetical fix linter error fix circular import issue
No functional changes intended. PiperOrigin-RevId: 483413031
PiperOrigin-RevId: 483425197
PiperOrigin-RevId: 483425722
overriden -> overridden
PiperOrigin-RevId: 483471278
PiperOrigin-RevId: 483490497
PiperOrigin-RevId: 483493703
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.