Code

Adding Jax tests for the M1 mac.#508

Code

Open

alxmrs wants to merge 37 commits intocubed-dev:maincubed-dev/cubed:mainfrom alxmrs:m1-jaxalxmrs/cubed:m1-jaxCopy head branch name to clipboard

+187-72

Commits

Commits on Sep 21, 2024

• gAdding Jax tests for the M1 mac.

  alxmrs

  committed
  90c65c6

• Providing flag to default to single precision instead of double for numpy create functions.

  Show description for f94d083
  alxmrs

  committed
  f94d083

• Matmul on jax-metal only supports floats and complex types. :/

  alxmrs

  committed
  5c10122

• Fixes for more tests.

  alxmrs

  committed
  83a752b

• Implemented feedback from Tom.

  alxmrs

  committed
  1f84ac8

• Added device argument.

  alxmrs

  committed
  b8445a0

• Exclude 'indexing' case.

  alxmrs

  committed
  61e51be

• Removing 'indexing' from default types for now.

  alxmrs

  committed
  a81be3d

• Fixed bag argument in test.

  alxmrs

  committed
  5dcab9f

• Random should respect dtypes.

  alxmrs

  committed
  4ce8e7a

• I think I got the linspace tests passing. Need to check if array equality tolerances are OK.

  alxmrs

  committed
  38d0b7b

• All jax tests pass locally.

  alxmrs

  committed
  e71fa63

• Cleanup for PR.

  alxmrs

  committed
  896a738

• Added developer instructions for getting set up on the M1+ processor.

  alxmrs

  committed
  a0080a9

• Fix env variable expressions.

  alxmrs

  committed
  5b00d82

• Looks like 14 specifically supports python 3.11 and up.

  alxmrs

  committed
  e9b53e3

• Splitting install steps for jax for mac and non-mac.

  alxmrs

  committed
  1ed8ac2

• Mac install now enables jaxlib compatibility.

  alxmrs

  committed
  72bd026

• Put the mac-specific env setting in the right place.

  alxmrs

  committed
  8b6b1f4

• Test the xlarge runner.

  alxmrs

  committed
  c4b51ad

• Extracted to method _to_default_precision.

  alxmrs

  committed
  7566849

• Added helpful note.

  alxmrs

  committed
  2f470d5

• No need to cast in the linspace test.

  alxmrs

  committed
  06e030d

• Remove macos-specific tests.

  alxmrs

  committed
  e642f8d

• Back tracking to previous macos build

  alxmrs

  committed
  2afba45

• Mac should use CPU Jax

  alxmrs

  committed
  38831b6

• Remove jax mac extra install.

  alxmrs

  committed
  9bd0210

• Tensordot testing with and without hardcoded types.

  alxmrs

  committed
  18e0aa0

• Defaulting to 64 bit precision.

  alxmrs

  committed
  4d132ff

• rm unneeded test arg.

  alxmrs

  committed
  9f6017a

• Revert tensordot change.

  alxmrs

  committed
  1449008

• Revert hardcoding precision as much as possible.

  alxmrs

  committed
  38dfc32

• Make precision a focus point in test matrix.

  alxmrs

  committed
  aebb2ab

• Fix underlying bug in asarray.

  alxmrs

  committed
  4687d33

• Exposing to_default_precision as a backend_array_api.

  alxmrs

  committed
  92a31cc

• All cubed tests pass!

  alxmrs

  committed
  dbefff4

• Inlined extracted function.

  Show description for b8ce6ec
  alxmrs

  committed
  b8ce6ec