search

ucl-bug / jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
GNU Lesser General Public License v3.0
118 stars 7 forks source link

Bump jaxlib from 0.1.75 to 0.1.76 #47

Closed dependabot[bot] closed 2 years ago

dependabot[bot] commented 2 years ago

Bumps jaxlib from 0.1.75 to 0.1.76.

Release notes

Sourced from jaxlib's releases.

Jaxlib v0.1.76

  • New features
    • Includes precompiled SASS for NVidia compute capability 8.0 GPUS (e.g. A100). Removes precompiled SASS for compute capability 6.1 so as not to increase the number of compute capabilities: GPUs with compute capability 6.1 can use the 6.0 SASS.
    • With jaxlib 0.1.76, JAX uses the MHLO MLIR dialect as its primary target compiler IR by default.
  • Breaking changes
    • Support for NumPy 1.18 has been dropped, per the deprecation policy. Please upgrade to a supported NumPy version.
  • Bug fixes
    • Fixed a bug where apparently identical pytreedef objects constructed by different routes do not compare as equal (#9066).
    • The JAX jit cache requires two static arguments to have identical types for a cache hit (#9311).
Changelog

Sourced from jaxlib's changelog.

jaxlib 0.1.76 (Jan 27, 2022)

  • New features
    • Includes precompiled SASS for NVidia compute capability 8.0 GPUS (e.g. A100). Removes precompiled SASS for compute capability 6.1 so as not to increase the number of compute capabilities: GPUs with compute capability 6.1 can use the 6.0 SASS.
    • With jaxlib 0.1.76, JAX uses the MHLO MLIR dialect as its primary target compiler IR by default.
  • Breaking changes
    • Support for NumPy 1.18 has been dropped, per the deprecation policy. Please upgrade to a supported NumPy version.
  • Bug fixes
    • Fixed a bug where apparently identical pytreedef objects constructed by different routes do not compare as equal (#9066).
    • The JAX jit cache requires two static arguments to have identical types for a cache hit (#9311).

jax 0.2.27 (Jan 18 2022)

  • GitHub commits.

  • Breaking changes:

    • Support for NumPy 1.18 has been dropped, per the deprecation policy. Please upgrade to a supported NumPy version.
    • The host_callback primitives have been simplified to drop the special autodiff handling for hcb.id_tap and id_print. From now on, only the primals are tapped. The old behavior can be obtained (for a limited time) by setting the JAX_HOST_CALLBACK_AD_TRANSFORMS environment variable, or the --flax_host_callback_ad_transforms flag. Additionally, added documentation for how to implement the old behavior using JAX custom AD APIs ({jax-issue}[#8678](https://github.com/google/jax/issues/8678)).
    • Sorting now matches the behavior of NumPy for 0.0 and NaN regardless of the bit representation. In particular, 0.0 and -0.0 are now treated as equivalent, where previously -0.0 was treated as less than 0.0. Additionally all NaN representations are now treated as equivalent and sorted to the end of the array. Previously negative NaN values were sorted to the front of the array, and NaN values with different internal bit representations were not treated as equivalent, and were sorted according to those bit patterns ({jax-issue}[#9178](https://github.com/google/jax/issues/9178)).
    • {func}jax.numpy.unique now treats NaN values in the same way as np.unique in NumPy versions 1.21 and newer: at most one NaN value will appear in the uniquified output ({jax-issue}9184).

  • Bug fixes:

    • host_callback now supports ad_checkpoint.checkpoint ({jax-issue}[#8907](https://github.com/google/jax/issues/8907)).

  • New features:

    • add jax.block_until_ready ({jax-issue}`#8941)
    • Added a new debugging flag/environment variable JAX_DUMP_IR_TO=/path. If set, JAX dumps the MHLO/HLO IR it generates for each computation to a

... (truncated)

Commits
  • 18a6614 Merge pull request #9353 from hawkinsp:jaxlib
  • be2f6a9 Update XLA for jaxlib 0.1.76 release.
  • 71468c6 Merge pull request #9352 from hawkinsp:flake
  • c54bb06 Fix flaky testcase LaxTest.testConvTransposePaddingList on GPU.
  • 6bd9292 Merge pull request #9343 from jakevdp:with-config
  • 1474f04 Explicit raise when comparing GDA equality.
  • b5dace5 Merge pull request #9331 from pschuh:id-assignment
  • f75c0f6 Merge pull request #9339 from jakevdp:rank-promotion-flag
  • 41b484c Merge pull request #9319 from jakevdp:doc-pre-commt
  • 1de3999 test_util: export with_config
  • Additional commits viewable in compare view


Dependabot compatibility score

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

Dependabot commands and options
You can trigger Dependabot actions by commenting on this PR: - `@dependabot rebase` will rebase this PR - `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it - `@dependabot merge` will merge this PR after your CI passes on it - `@dependabot squash and merge` will squash and merge this PR after your CI passes on it - `@dependabot cancel merge` will cancel a previously requested merge and block automerging - `@dependabot reopen` will reopen this PR if it is closed - `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually - `@dependabot ignore this major version` will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this minor version` will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this dependency` will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)
codecov[bot] commented 2 years ago

Codecov Report

Merging #47 (f49dec0) into main (70b7e26) will not change coverage. The diff coverage is n/a.

Impacted file tree graph

@@           Coverage Diff           @@
##             main      #47   +/-   ##
=======================================
  Coverage   49.78%   49.78%           
=======================================
  Files          10       10           
  Lines        1418     1418           
=======================================
  Hits          706      706           
  Misses        712      712

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 70b7e26...f49dec0. Read the comment docs.

dependabot[bot] commented 2 years ago

Superseded by #55.