Skip to content

Commit 789d7b5

Browse files
authored
Merge pull request #331 from ahrefs:threefry4x32
Threefry4x32 and Uint4x32-to-uniform operator support Tests are still expected to fail unfortunately.
2 parents 8335a27 + 633c613 commit 789d7b5

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

59 files changed

+1930
-755
lines changed

CHANGES.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
- This way we avoid introducing multiple operations on bits. We might yet split this operation into more fine grained ones.
1111
- TODO: Support of counter-based randomness, and behind-the-scenes key splitting by the relevant operations.
1212
- TODO: The cascade of splits uses the Tnode id, the train step and the tensor cell position.
13-
- TODO: Added a new operation `Uint4x32_to_prec_uniform`. It is added to `dedicated_access` and thus to `fetch_op` -- it cannot be `unop` because it cannot be implemented pointwise / is not shape-preserving due to efficient use of the 128 bits.
13+
- TODO: Added a new operation `Uint4x32_to_prec_uniform` to `unop`. It has a dedicated support in shape and projection inference to account for not being pointwise / is not shape-preserving due to efficient use of the 128 bits.
1414
- Added a field `params` to `Tensor.t` since we need to track parameters to properly initialize computations (see below).
1515

1616
### Changed

arrayjit/lib/anatomy_of_a_backend.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
The modules and files of `arrayjit` can loosely be divided into three parts.
2121

2222
- "Utility": generic code providing support across the project, both for `arrayjit` and `neural_nets_lib`.
23-
- The modules `Utils`, `Rand` and `Ppx_helper`.
23+
- The modules `Utils` and `Ppx_helper`.
2424
- "Frontend": the parts that are user-visible, concrete, and shared across the project.
2525
- `Task`: a wrapper for work execution (`unit -> unit`).
2626
- `Ops`: numeric precision specification types, primitive numerical operations.

0 commit comments

Comments
 (0)