2025-06-27 14:00.48: New job: test ahrefs/ocannl https://github.com/ahrefs/ocannl.git#refs/heads/master (f7224a34aa7346e5c1eaa29241a76124fc4c00be) (linux-x86_64:fedora-42-5.3_opam-2.3)
Base: ocaml/opam:fedora-42-ocaml-5.3@sha256:ca935fbf20f8d83e78dff4d973011a91d32565d7d98af772da42600dd103623c
Opam project build
To reproduce locally:
git clone --recursive "https://github.com/ahrefs/ocannl.git" -b "master" && cd "ocannl" && git reset --hard f7224a34
cat > Dockerfile <<'END-OF-DOCKERFILE'
FROM ocaml/opam:fedora-42-ocaml-5.3@sha256:ca935fbf20f8d83e78dff4d973011a91d32565d7d98af772da42600dd103623c
# fedora-42-5.3_opam-2.3
USER 1000:1000
ENV CLICOLOR_FORCE="1"
ENV OPAMCOLOR="always"
WORKDIR /src
RUN sudo dnf install -y findutils
RUN sudo ln -f /usr/bin/opam-2.3 /usr/bin/opam
RUN opam init --reinit -ni
RUN uname -rs && opam exec -- ocaml -version && opam --version
WORKDIR /src
RUN sudo chown opam /src
RUN cd ~/opam-repository && (git cat-file -e 7210227672addfb4db789560e45923f54c0aeab1 || git fetch origin master) && git reset -q --hard 7210227672addfb4db789560e45923f54c0aeab1 && git log --no-decorate -n1 --oneline && opam update -u
COPY --chown=1000:1000 neural_nets_lib.opam arrayjit.opam ./
RUN opam pin add -yn neural_nets_lib.dev './' && \
opam pin add -yn arrayjit.dev './'
RUN echo '(lang dune 3.0)' > './dune-project'
ENV DEPS="angstrom.0.16.1 astring.0.8.5 backoff.0.1.1 base.v0.17.3 base-bigarray.base base-bytes.base base-domains.base base-effects.base base-nnp.base base-threads.base base-unix.base bigarray-compat.1.1.0 bigstringaf.0.10.0 camlp-streams.5.0.1 camlzip.1.13 cmdliner.1.3.0 conf-libcurl.2 conf-libffi.2.0.0 conf-pkg-config.4 conf-zlib.1 cppo.1.8.0 csexp.1.5.2 csv.2.4 ctypes.0.23.0 ctypes-foreign.0.23.0 curl.0.10.0 dune.3.19.1 dune-configurator.3.19.1 fieldslib.v0.17.0 fmt.0.10.0 integers.0.7.0 jane-street-headers.v0.17.0 jst-config.v0.17.0 logs.0.8.0 mdx.2.5.0 mtime.2.1.0 multicore-magic.2.3.1 num.1.6 ocaml.5.3.0 ocaml-base-compiler.5.3.0 ocaml-compiler.5.3.0 ocaml-compiler-libs.v0.17.0 ocaml-config.3 ocaml-options-vanilla.1 ocaml-syntax-shims.1.0.0 ocaml-version.4.0.1 ocaml_intrinsics_kernel.v0.17.1 ocamlbuild.0.16.1 ocamlfind.1.9.8 parsexp.v0.17.0 pprint.20230830 ppx_assert.v0.17.0 ppx_base.v0.17.0 ppx_cold.v0.17.0 ppx_compare.v0.17.0 ppx_derivers.1.2.1 ppx_deriving.6.1.0 ppx_enumerate.v0.17.0 ppx_expect.v0.17.3 ppx_fields_conv.v0.17.0 ppx_globalize.v0.17.2 ppx_hash.v0.17.0 ppx_here.v0.17.0 ppx_inline_test.v0.17.1 ppx_minidebug.2.3.0 ppx_optcomp.v0.17.1 ppx_sexp_conv.v0.17.1 ppx_string.v0.17.0 ppx_variants_conv.v0.17.1 ppxlib.0.36.0 ppxlib_jane.v0.17.4 printbox.0.12 printbox-ext-plot.0.12 printbox-html.0.12 printbox-md.0.12 printbox-text.0.12 ptime.1.2.0 re.1.12.0 result.1.5 saturn_lockfree.0.5.0 seq.base sexplib.v0.17.0 sexplib0.v0.17.0 stdio.v0.17.0 stdlib-shims.0.3.0 thread-local-storage.0.2 time_now.v0.17.0 topkg.1.0.8 tyxml.4.6.0 uucp.16.0.0 uutf.1.0.4 variantslib.v0.17.0"
ENV CI="true"
ENV OCAMLCI="true"
RUN opam update --depexts && opam install --cli=2.3 --depext-only -y neural_nets_lib.dev arrayjit.dev $DEPS
RUN opam install $DEPS
COPY --chown=1000:1000 . /src
RUN opam exec -- dune build @install @check @runtest && rm -rf _build
END-OF-DOCKERFILE
docker build .
END-REPRO-BLOCK
2025-06-27 14:00.48: Using cache hint "ahrefs/ocannl-ocaml/opam:fedora-42-ocaml-5.3@sha256:ca935fbf20f8d83e78dff4d973011a91d32565d7d98af772da42600dd103623c-fedora-42-5.3_opam-2.3-82912e70cd030e6e6af979ac4425a6a2"
2025-06-27 14:00.48: Using OBuilder spec:
((from ocaml/opam:fedora-42-ocaml-5.3@sha256:ca935fbf20f8d83e78dff4d973011a91d32565d7d98af772da42600dd103623c)
(comment fedora-42-5.3_opam-2.3)
(user (uid 1000) (gid 1000))
(env CLICOLOR_FORCE 1)
(env OPAMCOLOR always)
(workdir /src)
(run (network host)
(shell "sudo dnf install -y findutils"))
(run (shell "sudo ln -f /usr/bin/opam-2.3 /usr/bin/opam"))
(run (shell "opam init --reinit -ni"))
(run (shell "uname -rs && opam exec -- ocaml -version && opam --version"))
(workdir /src)
(run (shell "sudo chown opam /src"))
(run (cache (opam-archives (target /home/opam/.opam/download-cache)))
(network host)
(shell "cd ~/opam-repository && (git cat-file -e 7210227672addfb4db789560e45923f54c0aeab1 || git fetch origin master) && git reset -q --hard 7210227672addfb4db789560e45923f54c0aeab1 && git log --no-decorate -n1 --oneline && opam update -u"))
(copy (src neural_nets_lib.opam arrayjit.opam) (dst ./))
(run (network host)
(shell "opam pin add -yn neural_nets_lib.dev './' && \
\nopam pin add -yn arrayjit.dev './'"))
(run (network host)
(shell "echo '(lang dune 3.0)' > './dune-project'"))
(env DEPS "angstrom.0.16.1 astring.0.8.5 backoff.0.1.1 base.v0.17.3 base-bigarray.base base-bytes.base base-domains.base base-effects.base base-nnp.base base-threads.base base-unix.base bigarray-compat.1.1.0 bigstringaf.0.10.0 camlp-streams.5.0.1 camlzip.1.13 cmdliner.1.3.0 conf-libcurl.2 conf-libffi.2.0.0 conf-pkg-config.4 conf-zlib.1 cppo.1.8.0 csexp.1.5.2 csv.2.4 ctypes.0.23.0 ctypes-foreign.0.23.0 curl.0.10.0 dune.3.19.1 dune-configurator.3.19.1 fieldslib.v0.17.0 fmt.0.10.0 integers.0.7.0 jane-street-headers.v0.17.0 jst-config.v0.17.0 logs.0.8.0 mdx.2.5.0 mtime.2.1.0 multicore-magic.2.3.1 num.1.6 ocaml.5.3.0 ocaml-base-compiler.5.3.0 ocaml-compiler.5.3.0 ocaml-compiler-libs.v0.17.0 ocaml-config.3 ocaml-options-vanilla.1 ocaml-syntax-shims.1.0.0 ocaml-version.4.0.1 ocaml_intrinsics_kernel.v0.17.1 ocamlbuild.0.16.1 ocamlfind.1.9.8 parsexp.v0.17.0 pprint.20230830 ppx_assert.v0.17.0 ppx_base.v0.17.0 ppx_cold.v0.17.0 ppx_compare.v0.17.0 ppx_derivers.1.2.1 ppx_deriving.6.1.0 ppx_enumerate.v0.17.0 ppx_expect.v0.17.3 ppx_fields_conv.v0.17.0 ppx_globalize.v0.17.2 ppx_hash.v0.17.0 ppx_here.v0.17.0 ppx_inline_test.v0.17.1 ppx_minidebug.2.3.0 ppx_optcomp.v0.17.1 ppx_sexp_conv.v0.17.1 ppx_string.v0.17.0 ppx_variants_conv.v0.17.1 ppxlib.0.36.0 ppxlib_jane.v0.17.4 printbox.0.12 printbox-ext-plot.0.12 printbox-html.0.12 printbox-md.0.12 printbox-text.0.12 ptime.1.2.0 re.1.12.0 result.1.5 saturn_lockfree.0.5.0 seq.base sexplib.v0.17.0 sexplib0.v0.17.0 stdio.v0.17.0 stdlib-shims.0.3.0 thread-local-storage.0.2 time_now.v0.17.0 topkg.1.0.8 tyxml.4.6.0 uucp.16.0.0 uutf.1.0.4 variantslib.v0.17.0")
(env CI true)
(env OCAMLCI true)
(run (cache (opam-archives (target /home/opam/.opam/download-cache)))
(network host)
(shell "opam update --depexts && opam install --cli=2.3 --depext-only -y neural_nets_lib.dev arrayjit.dev $DEPS"))
(run (cache (opam-archives (target /home/opam/.opam/download-cache)))
(network host)
(shell "opam install $DEPS"))
(copy (src .) (dst /src))
(run (shell "opam exec -- dune build @install @check @runtest && rm -rf _build"))
)
2025-06-27 14:00.48: Waiting for resource in pool OCluster
2025-06-27 14:00.48: Waiting for worker…
2025-06-27 14:00.48: Got resource from pool OCluster
Building on toxis.caelum.ci.dev
All commits already cached
HEAD is now at f7224a34 By Claude Sonnet: fix missing Affine expansion during inlining, document low_level.ml optimizations
(from ocaml/opam:fedora-42-ocaml-5.3@sha256:ca935fbf20f8d83e78dff4d973011a91d32565d7d98af772da42600dd103623c)
2025-06-27 14:00.52 ---> using "6219cec5618da56ab7f2d1a249180f4afe319ecdd103db509824f628c646fc56" from cache
/: (comment fedora-42-5.3_opam-2.3)
/: (user (uid 1000) (gid 1000))
/: (env CLICOLOR_FORCE 1)
/: (env OPAMCOLOR always)
/: (workdir /src)
/src: (run (network host)
(shell "sudo dnf install -y findutils"))
Updating and loading repositories:
Fedora 42 openh264 (From Cisco) - x86_ 100% | 3.9 KiB/s | 986.0 B | 00m00s
Fedora 42 - x86_64 - Updates 100% | 85.7 KiB/s | 10.5 KiB | 00m00s
Fedora 42 - x86_64 100% | 200.6 KiB/s | 26.9 KiB | 00m00s
Fedora 42 openh264 (From Cisco) - x86_ 100% | 17.7 KiB/s | 5.8 KiB | 00m00s
Fedora 42 - x86_64 - Updates 100% | 4.2 MiB/s | 4.2 MiB | 00m01s
Repositories loaded.
Package "findutils-1:4.10.0-5.fc42.x86_64" is already installed.
Nothing to do.
2025-06-27 14:00.52 ---> using "b1f0db1fada5631e0e8a089e4b21d3aba9fcebb8766eb6c5a9e6a096ca963ec7" from cache
/src: (run (shell "sudo ln -f /usr/bin/opam-2.3 /usr/bin/opam"))
2025-06-27 14:00.52 ---> using "36f20adfdb7a1fd3920d73b330c5ff2e9a27026c342876b307bcafc268e4eecd" from cache
/src: (run (shell "opam init --reinit -ni"))
Configuring from /home/opam/.opamrc and then from built-in defaults.
Checking for available remotes: rsync and local, git.
- you won't be able to use mercurial repositories unless you install the hg command on your system.
- you won't be able to use darcs repositories unless you install the darcs command on your system.
This development version of opam requires an update to the layout of /home/opam/.opam from version 2.0 to version 2.2, which can't be reverted.
You may want to back it up before going further.
Continue? [y/n] y
Format upgrade done.
<><> Updating repositories ><><><><><><><><><><><><><><><><><><><><><><><><><><>
[default] Initialised
2025-06-27 14:00.52 ---> using "183296d1720211aca3f522267709abbd22789de72e87a55022fd39b32feff80b" from cache
/src: (run (shell "uname -rs && opam exec -- ocaml -version && opam --version"))
Linux 5.15.0-134-generic
The OCaml toplevel, version 5.3.0
2.3.0
2025-06-27 14:00.52 ---> using "626b7abede693a100c736148e8f886bf78f49cecf8d70aa67da157cd49d141e7" from cache
/src: (workdir /src)
/src: (run (shell "sudo chown opam /src"))
2025-06-27 14:00.52 ---> using "4994b2b063705049a578a29c9bf8c120bb196e13f645e32fb92443c24d9db643" from cache
/src: (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
(network host)
(shell "cd ~/opam-repository && (git cat-file -e 7210227672addfb4db789560e45923f54c0aeab1 || git fetch origin master) && git reset -q --hard 7210227672addfb4db789560e45923f54c0aeab1 && git log --no-decorate -n1 --oneline && opam update -u"))
From https://github.com/ocaml/opam-repository
* branch master -> FETCH_HEAD
bc726805e3..2ca36230be master -> origin/master
7210227672 Merge pull request #28100 from smuenzel/ppx-deriving-6.1.0-ocaml-5.4
<><> Updating package repositories ><><><><><><><><><><><><><><><><><><><><><><>
[default] synchronised from git+file:///home/opam/opam-repository
Everything as up-to-date as possible (run with --verbose to show unavailable upgrades).
However, you may "opam upgrade" these packages explicitly, which will ask permission to downgrade or uninstall the conflicting packages.
Nothing to do.
# To update the current shell environment, run: eval $(opam env)
2025-06-27 14:00.52 ---> using "3a2c7f3d7f858cbf5495fa9633d9d7ab56096a2d0bdf3152c92a7cc6bef114ec" from cache
/src: (copy (src neural_nets_lib.opam arrayjit.opam) (dst ./))
2025-06-27 14:00.52 ---> using "dd5166dc96c0d3449f9011e1d22452fe1f252ba83b18008ec8c6b261380b79f9" from cache
/src: (run (network host)
(shell "opam pin add -yn neural_nets_lib.dev './' && \
\nopam pin add -yn arrayjit.dev './'"))
[neural_nets_lib.dev] synchronised (file:///src)
neural_nets_lib is now pinned to file:///src (version dev)
[arrayjit.dev] synchronised (file:///src)
arrayjit is now pinned to file:///src (version dev)
2025-06-27 14:00.52 ---> using "d8bd79ed7ec398484a3126847d77f7404e5dc6df25da16ff537d70577bdcbf99" from cache
/src: (run (network host)
(shell "echo '(lang dune 3.0)' > './dune-project'"))
2025-06-27 14:00.52 ---> using "9cfaa294eee11760275e61355d8240df0f21f68fe32e5dbbd0b66ae4f6ef99c1" from cache
/src: (env DEPS "angstrom.0.16.1 astring.0.8.5 backoff.0.1.1 base.v0.17.3 base-bigarray.base base-bytes.base base-domains.base base-effects.base base-nnp.base base-threads.base base-unix.base bigarray-compat.1.1.0 bigstringaf.0.10.0 camlp-streams.5.0.1 camlzip.1.13 cmdliner.1.3.0 conf-libcurl.2 conf-libffi.2.0.0 conf-pkg-config.4 conf-zlib.1 cppo.1.8.0 csexp.1.5.2 csv.2.4 ctypes.0.23.0 ctypes-foreign.0.23.0 curl.0.10.0 dune.3.19.1 dune-configurator.3.19.1 fieldslib.v0.17.0 fmt.0.10.0 integers.0.7.0 jane-street-headers.v0.17.0 jst-config.v0.17.0 logs.0.8.0 mdx.2.5.0 mtime.2.1.0 multicore-magic.2.3.1 num.1.6 ocaml.5.3.0 ocaml-base-compiler.5.3.0 ocaml-compiler.5.3.0 ocaml-compiler-libs.v0.17.0 ocaml-config.3 ocaml-options-vanilla.1 ocaml-syntax-shims.1.0.0 ocaml-version.4.0.1 ocaml_intrinsics_kernel.v0.17.1 ocamlbuild.0.16.1 ocamlfind.1.9.8 parsexp.v0.17.0 pprint.20230830 ppx_assert.v0.17.0 ppx_base.v0.17.0 ppx_cold.v0.17.0 ppx_compare.v0.17.0 ppx_derivers.1.2.1 ppx_deriving.6.1.0 ppx_enumerate.v0.17.0 ppx_expect.v0.17.3 ppx_fields_conv.v0.17.0 ppx_globalize.v0.17.2 ppx_hash.v0.17.0 ppx_here.v0.17.0 ppx_inline_test.v0.17.1 ppx_minidebug.2.3.0 ppx_optcomp.v0.17.1 ppx_sexp_conv.v0.17.1 ppx_string.v0.17.0 ppx_variants_conv.v0.17.1 ppxlib.0.36.0 ppxlib_jane.v0.17.4 printbox.0.12 printbox-ext-plot.0.12 printbox-html.0.12 printbox-md.0.12 printbox-text.0.12 ptime.1.2.0 re.1.12.0 result.1.5 saturn_lockfree.0.5.0 seq.base sexplib.v0.17.0 sexplib0.v0.17.0 stdio.v0.17.0 stdlib-shims.0.3.0 thread-local-storage.0.2 time_now.v0.17.0 topkg.1.0.8 tyxml.4.6.0 uucp.16.0.0 uutf.1.0.4 variantslib.v0.17.0")
/src: (env CI true)
/src: (env OCAMLCI true)
/src: (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
(network host)
(shell "opam update --depexts && opam install --cli=2.3 --depext-only -y neural_nets_lib.dev arrayjit.dev $DEPS"))
+ /usr/sbin/sudo "yum" "makecache"
- Updating and loading repositories:
- Repositories loaded.
- Metadata cache created.
<><> Synchronising pinned packages ><><><><><><><><><><><><><><><><><><><><><><>
[neural_nets_lib.dev] synchronised (file:///src)
[arrayjit.dev] synchronised (file:///src)
[NOTE] Package ocaml-options-vanilla is already installed (current version is 1).
[NOTE] Package ocaml-config is already installed (current version is 3).
[NOTE] Package ocaml-compiler is already installed (current version is 5.3.0).
[NOTE] Package ocaml-base-compiler is already installed (current version is 5.3.0).
[NOTE] Package ocaml is already installed (current version is 5.3.0).
[NOTE] Package base-unix is already installed (current version is base).
[NOTE] Package base-threads is already installed (current version is base).
[NOTE] Package base-nnp is already installed (current version is base).
[NOTE] Package base-effects is already installed (current version is base).
[NOTE] Package base-domains is already installed (current version is base).
[NOTE] Package base-bigarray is already installed (current version is base).
The following system packages will first need to be installed:
libcurl-devel libffi-devel zlib-ng-compat-devel
<><> Handling external dependencies <><><><><><><><><><><><><><><><><><><><><><>
+ /usr/sbin/sudo "yum" "install" "-y" "libcurl-devel" "libffi-devel" "zlib-ng-compat-devel"
- Updating and loading repositories:
- Repositories loaded.
- Package Arch Version Repository Size
- Upgrading:
- krb5-libs x86_64 1.21.3-6.fc42 updates 2.3 MiB
- replacing krb5-libs x86_64 1.21.3-5.fc42 b67228ed38864378b0e1f36b75127eab 2.3 MiB
- libcurl x86_64 8.11.1-5.fc42 updates 834.0 KiB
- replacing libcurl x86_64 8.11.1-4.fc42 b67228ed38864378b0e1f36b75127eab 842.1 KiB
- libssh x86_64 0.11.2-1.fc42 updates 566.7 KiB
- replacing libssh x86_64 0.11.1-4.fc42 b67228ed38864378b0e1f36b75127eab 565.5 KiB
- libssh-config noarch 0.11.2-1.fc42 updates 277.0 B
- replacing libssh-config noarch 0.11.1-4.fc42 b67228ed38864378b0e1f36b75127eab 277.0 B
- Installing:
- libcurl-devel x86_64 8.11.1-5.fc42 updates 1.3 MiB
- libffi-devel x86_64 3.4.6-5.fc42 fedora 33.1 KiB
- zlib-ng-compat-devel x86_64 2.2.4-3.fc42 fedora 107.0 KiB
- Installing dependencies:
- brotli x86_64 1.1.0-6.fc42 fedora 31.6 KiB
- brotli-devel x86_64 1.1.0-6.fc42 fedora 65.6 KiB
- cmake-filesystem x86_64 3.31.6-2.fc42 fedora 0.0 B
- keyutils-libs-devel x86_64 1.6.3-5.fc42 fedora 48.2 KiB
- krb5-devel x86_64 1.21.3-6.fc42 updates 705.9 KiB
- libcom_err-devel x86_64 1.47.2-3.fc42 fedora 16.7 KiB
- libidn2-devel x86_64 2.3.8-1.fc42 fedora 149.1 KiB
- libkadm5 x86_64 1.21.3-6.fc42 updates 213.9 KiB
- libnghttp2-devel x86_64 1.64.0-3.fc42 fedora 295.4 KiB
- libpsl-devel x86_64 0.21.5-5.fc42 fedora 110.3 KiB
- libselinux-devel x86_64 3.8-2.fc42 updates 126.8 KiB
- libsepol-devel x86_64 3.8-1.fc42 fedora 120.8 KiB
- libssh-devel x86_64 0.11.2-1.fc42 updates 178.0 KiB
- libverto-devel x86_64 0.3.2-10.fc42 fedora 25.7 KiB
- openssl-devel x86_64 1:3.2.4-3.fc42 fedora 4.3 MiB
- pcre2-devel x86_64 10.45-1.fc42 fedora 2.1 MiB
- pcre2-utf16 x86_64 10.45-1.fc42 fedora 626.3 KiB
- pcre2-utf32 x86_64 10.45-1.fc42 fedora 598.2 KiB
- publicsuffix-list noarch 20250616-1.fc42 updates 332.8 KiB
-
- Transaction Summary:
- Installing: 22 packages
- Upgrading: 4 packages
- Replacing: 4 packages
-
- Total size of inbound packages is 7 MiB. Need to download 7 MiB.
- After this operation, 11 MiB extra will be used (install 15 MiB, remove 4 MiB).
- [ 1/26] libffi-devel-0:3.4.6-5.fc42.x86 100% | 239.8 KiB/s | 28.8 KiB | 00m00s
- [ 2/26] libcurl-devel-0:8.11.1-5.fc42.x 100% | 4.8 MiB/s | 881.9 KiB | 00m00s
- [ 3/26] zlib-ng-compat-devel-0:2.2.4-3. 100% | 214.7 KiB/s | 38.4 KiB | 00m00s
- [ 4/26] krb5-devel-0:1.21.3-6.fc42.x86_ 100% | 3.6 MiB/s | 142.5 KiB | 00m00s
- [ 5/26] libssh-devel-0:0.11.2-1.fc42.x8 100% | 2.4 MiB/s | 41.5 KiB | 00m00s
- [ 6/26] cmake-filesystem-0:3.31.6-2.fc4 100% | 146.6 KiB/s | 17.6 KiB | 00m00s
- [ 7/26] brotli-0:1.1.0-6.fc42.x86_64 100% | 2.4 MiB/s | 19.9 KiB | 00m00s
- [ 8/26] libkadm5-0:1.21.3-6.fc42.x86_64 100% | 741.5 KiB/s | 76.4 KiB | 00m00s
- [ 9/26] brotli-devel-0:1.1.0-6.fc42.x86 100% | 424.1 KiB/s | 33.9 KiB | 00m00s
- [10/26] libnghttp2-devel-0:1.64.0-3.fc4 100% | 502.2 KiB/s | 55.7 KiB | 00m00s
- [11/26] libpsl-devel-0:0.21.5-5.fc42.x8 100% | 200.0 KiB/s | 33.2 KiB | 00m00s
- [12/26] libidn2-devel-0:2.3.8-1.fc42.x8 100% | 268.8 KiB/s | 64.0 KiB | 00m00s
- [13/26] libsepol-devel-0:3.8-1.fc42.x86 100% | 1.4 MiB/s | 48.4 KiB | 00m00s
- [14/26] libselinux-devel-0:3.8-2.fc42.x 100% | 3.4 MiB/s | 151.5 KiB | 00m00s
- [15/26] libcom_err-devel-0:1.47.2-3.fc4 100% | 214.6 KiB/s | 16.7 KiB | 00m00s
- [16/26] keyutils-libs-devel-0:1.6.3-5.f 100% | 453.9 KiB/s | 59.9 KiB | 00m00s
- [17/26] libverto-devel-0:0.3.2-10.fc42. 100% | 208.7 KiB/s | 14.4 KiB | 00m00s
- [18/26] publicsuffix-list-0:20250616-1. 100% | 4.9 MiB/s | 89.9 KiB | 00m00s
- [19/26] pcre2-utf16-0:10.45-1.fc42.x86_ 100% | 2.3 MiB/s | 241.9 KiB | 00m00s
- [20/26] openssl-devel-1:3.2.4-3.fc42.x8 100% | 7.1 MiB/s | 2.8 MiB | 00m00s
- [21/26] libcurl-0:8.11.1-5.fc42.x86_64 100% | 9.3 MiB/s | 371.7 KiB | 00m00s
- [22/26] pcre2-utf32-0:10.45-1.fc42.x86_ 100% | 2.3 MiB/s | 228.7 KiB | 00m00s
- [23/26] krb5-libs-0:1.21.3-6.fc42.x86_6 100% | 18.1 MiB/s | 759.8 KiB | 00m00s
- [24/26] libssh-config-0:0.11.2-1.fc42.n 100% | 553.2 KiB/s | 8.9 KiB | 00m00s
- [25/26] pcre2-devel-0:10.45-1.fc42.x86_ 100% | 2.2 MiB/s | 543.4 KiB | 00m00s
- [26/26] libssh-0:0.11.2-1.fc42.x86_64 100% | 5.3 MiB/s | 232.5 KiB | 00m00s
- --------------------------------------------------------------------------------
- [26/26] Total 100% | 5.8 MiB/s | 6.9 MiB | 00m01s
- Running transaction
- [ 1/32] Verify package files 100% | 236.0 B/s | 26.0 B | 00m00s
- [ 2/32] Prepare transaction 100% | 105.0 B/s | 30.0 B | 00m00s
- [ 3/32] Upgrading krb5-libs-0:1.21.3-6. 100% | 65.5 MiB/s | 2.3 MiB | 00m00s
- [ 4/32] Installing cmake-filesystem-0:3 100% | 1.2 MiB/s | 7.6 KiB | 00m00s
- [ 5/32] Installing zlib-ng-compat-devel 100% | 26.5 MiB/s | 108.5 KiB | 00m00s
- [ 6/32] Installing libkadm5-0:1.21.3-6. 100% | 35.1 MiB/s | 215.9 KiB | 00m00s
- [ 7/32] Upgrading libssh-config-0:0.11. 100% | 398.4 KiB/s | 816.0 B | 00m00s
- [ 8/32] Upgrading libssh-0:0.11.2-1.fc4 100% | 55.5 MiB/s | 568.8 KiB | 00m00s
- [ 9/32] Installing libssh-devel-0:0.11. 100% | 35.3 MiB/s | 180.6 KiB | 00m00s
- [10/32] Upgrading libcurl-0:8.11.1-5.fc 100% | 58.3 MiB/s | 835.1 KiB | 00m00s
- [11/32] Installing pcre2-utf32-0:10.45- 100% | 65.0 MiB/s | 599.1 KiB | 00m00s
- [12/32] Installing pcre2-utf16-0:10.45- 100% | 61.2 MiB/s | 627.1 KiB | 00m00s
- [13/32] Installing pcre2-devel-0:10.45- 100% | 46.5 MiB/s | 2.1 MiB | 00m00s
- [14/32] Installing publicsuffix-list-0: 100% | 54.3 MiB/s | 333.8 KiB | 00m00s
- [15/32] Installing libpsl-devel-0:0.21. 100% | 37.0 MiB/s | 113.6 KiB | 00m00s
- [16/32] Installing libverto-devel-0:0.3 100% | 12.9 MiB/s | 26.4 KiB | 00m00s
- [17/32] Installing libcom_err-devel-0:1 100% | 522.1 KiB/s | 18.3 KiB | 00m00s
- [18/32] Installing keyutils-libs-devel- 100% | 18.0 MiB/s | 55.2 KiB | 00m00s
- [19/32] Installing libsepol-devel-0:3.8 100% | 13.9 MiB/s | 128.3 KiB | 00m00s
- [20/32] Installing libselinux-devel-0:3 100% | 19.7 MiB/s | 161.6 KiB | 00m00s
- [21/32] Installing krb5-devel-0:1.21.3- 100% | 10.3 MiB/s | 715.2 KiB | 00m00s
- [22/32] Installing openssl-devel-1:3.2. 100% | 48.5 MiB/s | 5.2 MiB | 00m00s
- [23/32] Installing libnghttp2-devel-0:1 100% | 57.9 MiB/s | 296.5 KiB | 00m00s
- [24/32] Installing libidn2-devel-0:2.3. 100% | 38.3 MiB/s | 156.7 KiB | 00m00s
- [25/32] Installing brotli-0:1.1.0-6.fc4 100% | 949.9 KiB/s | 32.3 KiB | 00m00s
- [26/32] Installing brotli-devel-0:1.1.0 100% | 8.3 MiB/s | 68.0 KiB | 00m00s
- [27/32] Installing libcurl-devel-0:8.11 100% | 26.2 MiB/s | 1.4 MiB | 00m00s
- [28/32] Installing libffi-devel-0:3.4.6 100% | 17.0 MiB/s | 34.8 KiB | 00m00s
- [29/32] Removing libcurl-0:8.11.1-4.fc4 100% | 2.3 KiB/s | 7.0 B | 00m00s
- [30/32] Removing libssh-0:0.11.1-4.fc42 100% | 3.4 KiB/s | 14.0 B | 00m00s
- [31/32] Removing libssh-config-0:0.11.1 100% | 2.9 KiB/s | 3.0 B | 00m00s
- [32/32] Removing krb5-libs-0:1.21.3-5.f 100% | 392.0 B/s | 64.0 B | 00m00s
- Complete!
+ /usr/sbin/rpm "-q" "--whatprovides" "libcurl-devel" "libffi-devel" "zlib-ng-compat-devel"
- libcurl-devel-8.11.1-5.fc42.x86_64
- libffi-devel-3.4.6-5.fc42.x86_64
- zlib-ng-compat-devel-2.2.4-3.fc42.x86_64
2025-06-27 14:00.52 ---> using "a51d3d0e860bc2aca3554ec0d97927cffbb8fe0543cfa92a64b81658d03f7b78" from cache
/src: (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
(network host)
(shell "opam install $DEPS"))
[NOTE] Package ocaml-options-vanilla is already installed (current version is 1).
[NOTE] Package ocaml-config is already installed (current version is 3).
[NOTE] Package ocaml-compiler is already installed (current version is 5.3.0).
[NOTE] Package ocaml-base-compiler is already installed (current version is 5.3.0).
[NOTE] Package ocaml is already installed (current version is 5.3.0).
[NOTE] Package base-unix is already installed (current version is base).
[NOTE] Package base-threads is already installed (current version is base).
[NOTE] Package base-nnp is already installed (current version is base).
[NOTE] Package base-effects is already installed (current version is base).
[NOTE] Package base-domains is already installed (current version is base).
[NOTE] Package base-bigarray is already installed (current version is base).
The following actions will be performed:
=== install 81 packages
- install angstrom 0.16.1
- install astring 0.8.5
- install backoff 0.1.1
- install base v0.17.3
- install base-bytes base
- install bigarray-compat 1.1.0
- install bigstringaf 0.10.0
- install camlp-streams 5.0.1
- install camlzip 1.13
- install cmdliner 1.3.0
- install conf-libcurl 2
- install conf-libffi 2.0.0
- install conf-pkg-config 4
- install conf-zlib 1
- install cppo 1.8.0
- install csexp 1.5.2
- install csv 2.4
- install ctypes 0.23.0
- install ctypes-foreign 0.23.0
- install curl 0.10.0
- install dune 3.19.1
- install dune-configurator 3.19.1
- install fieldslib v0.17.0
- install fmt 0.10.0
- install integers 0.7.0
- install jane-street-headers v0.17.0
- install jst-config v0.17.0
- install logs 0.8.0
- install mdx 2.5.0
- install mtime 2.1.0
- install multicore-magic 2.3.1
- install num 1.6
- install ocaml-compiler-libs v0.17.0
- install ocaml-syntax-shims 1.0.0
- install ocaml-version 4.0.1
- install ocaml_intrinsics_kernel v0.17.1
- install ocamlbuild 0.16.1
- install ocamlfind 1.9.8
- install parsexp v0.17.0
- install pprint 20230830
- install ppx_assert v0.17.0
- install ppx_base v0.17.0
- install ppx_cold v0.17.0
- install ppx_compare v0.17.0
- install ppx_derivers 1.2.1
- install ppx_deriving 6.1.0
- install ppx_enumerate v0.17.0
- install ppx_expect v0.17.3
- install ppx_fields_conv v0.17.0
- install ppx_globalize v0.17.2
- install ppx_hash v0.17.0
- install ppx_here v0.17.0
- install ppx_inline_test v0.17.1
- install ppx_minidebug 2.3.0
- install ppx_optcomp v0.17.1
- install ppx_sexp_conv v0.17.1
- install ppx_string v0.17.0
- install ppx_variants_conv v0.17.1
- install ppxlib 0.36.0
- install ppxlib_jane v0.17.4
- install printbox 0.12
- install printbox-ext-plot 0.12
- install printbox-html 0.12
- install printbox-md 0.12
- install printbox-text 0.12
- install ptime 1.2.0
- install re 1.12.0
- install result 1.5
- install saturn_lockfree 0.5.0
- install seq base
- install sexplib v0.17.0
- install sexplib0 v0.17.0
- install stdio v0.17.0
- install stdlib-shims 0.3.0
- install thread-local-storage 0.2
- install time_now v0.17.0
- install topkg 1.0.8
- install tyxml 4.6.0
- install uucp 16.0.0
- install uutf 1.0.4
- install variantslib v0.17.0
<><> Processing actions <><><><><><><><><><><><><><><><><><><><><><><><><><><><>
-> retrieved backoff.0.1.1 (cached)
-> retrieved astring.0.8.5 (cached)
-> retrieved angstrom.0.16.1 (cached)
-> retrieved bigarray-compat.1.1.0 (cached)
-> retrieved base.v0.17.3 (cached)
-> retrieved bigstringaf.0.10.0 (cached)
-> retrieved camlp-streams.5.0.1 (cached)
-> retrieved camlzip.1.13 (cached)
-> retrieved cmdliner.1.3.0 (cached)
-> retrieved cppo.1.8.0 (cached)
-> installed conf-libcurl.2
-> installed conf-pkg-config.4
-> retrieved csexp.1.5.2 (cached)
-> retrieved csv.2.4 (cached)
-> retrieved ctypes.0.23.0, ctypes-foreign.0.23.0 (cached)
-> installed conf-libffi.2.0.0
-> installed conf-zlib.1
-> retrieved curl.0.10.0 (cached)
-> retrieved fieldslib.v0.17.0 (cached)
-> retrieved integers.0.7.0 (cached)
-> retrieved fmt.0.10.0 (cached)
-> retrieved jane-street-headers.v0.17.0 (cached)
-> retrieved jst-config.v0.17.0 (cached)
-> retrieved logs.0.8.0 (cached)
-> retrieved mtime.2.1.0 (cached)
-> retrieved mdx.2.5.0 (cached)
-> retrieved multicore-magic.2.3.1 (cached)
-> retrieved num.1.6 (cached)
-> retrieved ocaml-compiler-libs.v0.17.0 (cached)
-> retrieved ocaml-syntax-shims.1.0.0 (cached)
-> retrieved ocaml-version.4.0.1 (cached)
-> retrieved ocaml_intrinsics_kernel.v0.17.1 (cached)
-> retrieved ocamlbuild.0.16.1 (cached)
-> retrieved ocamlfind.1.9.8 (cached)
-> retrieved parsexp.v0.17.0 (cached)
-> retrieved pprint.20230830 (cached)
-> retrieved ppx_assert.v0.17.0 (cached)
-> retrieved ppx_base.v0.17.0 (cached)
-> retrieved ppx_cold.v0.17.0 (cached)
-> retrieved ppx_compare.v0.17.0 (cached)
-> retrieved ppx_derivers.1.2.1 (cached)
-> retrieved ppx_enumerate.v0.17.0 (cached)
-> retrieved ppx_deriving.6.1.0 (cached)
-> retrieved ppx_expect.v0.17.3 (cached)
-> retrieved ppx_fields_conv.v0.17.0 (cached)
-> retrieved ppx_globalize.v0.17.2 (cached)
-> retrieved ppx_hash.v0.17.0 (cached)
-> retrieved ppx_here.v0.17.0 (cached)
-> retrieved ppx_inline_test.v0.17.1 (cached)
-> retrieved ppx_optcomp.v0.17.1 (cached)
-> retrieved ppx_sexp_conv.v0.17.1 (cached)
-> retrieved dune.3.19.1, dune-configurator.3.19.1 (cached)
-> retrieved ppx_minidebug.2.3.0 (cached)
-> retrieved ppx_string.v0.17.0 (cached)
-> retrieved ppx_variants_conv.v0.17.1 (cached)
-> retrieved ppxlib_jane.v0.17.4 (cached)
-> retrieved ptime.1.2.0 (cached)
-> retrieved re.1.12.0 (cached)
-> installed cmdliner.1.3.0
-> installed num.1.6
-> retrieved result.1.5 (cached)
-> retrieved ppxlib.0.36.0 (cached)
-> retrieved seq.base (cached)
-> installed seq.base
-> retrieved saturn_lockfree.0.5.0 (cached)
-> retrieved sexplib.v0.17.0 (cached)
-> retrieved printbox.0.12, printbox-ext-plot.0.12, printbox-html.0.12, printbox-md.0.12, printbox-text.0.12 (cached)
-> retrieved sexplib0.v0.17.0 (cached)
-> retrieved stdio.v0.17.0 (cached)
-> retrieved stdlib-shims.0.3.0 (cached)
-> retrieved thread-local-storage.0.2 (cached)
-> retrieved time_now.v0.17.0 (cached)
-> retrieved topkg.1.0.8 (cached)
-> retrieved tyxml.4.6.0 (cached)
-> retrieved uutf.1.0.4 (cached)
-> retrieved variantslib.v0.17.0 (cached)
-> retrieved uucp.16.0.0 (cached)
-> installed ocamlbuild.0.16.1
-> installed ocamlfind.1.9.8
-> installed base-bytes.base
-> installed camlzip.1.13
-> installed topkg.1.0.8
-> installed mtime.2.1.0
-> installed uutf.1.0.4
-> installed fmt.0.10.0
-> installed ptime.1.2.0
-> installed astring.0.8.5
-> installed logs.0.8.0
-> installed dune.3.19.1
-> installed jane-street-headers.v0.17.0
-> installed ppx_derivers.1.2.1
-> installed csexp.1.5.2
-> installed backoff.0.1.1
-> installed bigarray-compat.1.1.0
-> installed camlp-streams.5.0.1
-> installed csv.2.4
-> installed multicore-magic.2.3.1
-> installed ocaml-version.4.0.1
-> installed ocaml_intrinsics_kernel.v0.17.1
-> installed pprint.20230830
-> installed printbox.0.12
-> installed result.1.5
-> installed sexplib0.v0.17.0
-> installed stdlib-shims.0.3.0
-> installed thread-local-storage.0.2
-> installed ocaml-syntax-shims.1.0.0
-> installed ocaml-compiler-libs.v0.17.0
-> installed cppo.1.8.0
-> installed re.1.12.0
-> installed integers.0.7.0
-> installed saturn_lockfree.0.5.0
-> installed dune-configurator.3.19.1
-> installed parsexp.v0.17.0
-> installed bigstringaf.0.10.0
-> installed sexplib.v0.17.0
-> installed angstrom.0.16.1
-> installed mdx.2.5.0
-> installed tyxml.4.6.0
-> installed printbox-html.0.12
-> installed curl.0.10.0
-> installed uucp.16.0.0
-> installed printbox-text.0.12
-> installed printbox-md.0.12
-> installed ctypes.0.23.0
-> installed printbox-ext-plot.0.12
-> installed base.v0.17.3
-> installed fieldslib.v0.17.0
-> installed variantslib.v0.17.0
-> installed stdio.v0.17.0
-> installed ctypes-foreign.0.23.0
-> installed ppxlib.0.36.0
-> installed ppxlib_jane.v0.17.4
-> installed ppx_optcomp.v0.17.1
-> installed ppx_here.v0.17.0
-> installed ppx_cold.v0.17.0
-> installed ppx_variants_conv.v0.17.1
-> installed ppx_fields_conv.v0.17.0
-> installed ppx_deriving.6.1.0
-> installed ppx_enumerate.v0.17.0
-> installed ppx_globalize.v0.17.2
-> installed ppx_compare.v0.17.0
-> installed ppx_sexp_conv.v0.17.1
-> installed ppx_hash.v0.17.0
-> installed ppx_assert.v0.17.0
-> installed ppx_base.v0.17.0
-> installed ppx_minidebug.2.3.0
-> installed jst-config.v0.17.0
-> installed ppx_string.v0.17.0
-> installed time_now.v0.17.0
-> installed ppx_inline_test.v0.17.1
-> installed ppx_expect.v0.17.3
Done.
# To update the current shell environment, run: eval $(opam env)
2025-06-27 14:00.52 ---> using "22f4aa9826c93953a9d195932d45b7588c483aa23d15a0dcfeeed4b6ac79d440" from cache
/src: (copy (src .) (dst /src))
2025-06-27 14:00.53 ---> saved as "942d70cbe5db95570ac5bc53ff0545c3fd577307905cd704d8e631d0dc5a836a"
/src: (run (shell "opam exec -- dune build @install @check @runtest && rm -rf _build"))
File "arrayjit/test/dune", line 6, characters 7-15:
6 | (pps ppx_jane)))
^^^^^^^^
Error: Library "ppx_jane" not found.
-> required by
_build/default/arrayjit/test/.merlin-conf/exe-test_numerical_types
-> required by alias arrayjit/test/check
(cd _build/default/test/config && ../../arrayjit/bin/read_config.exe --read=backend)
Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test/config/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Wrote value of 'backend' to ocannl_backend.txt
(cd _build/default/test_ppx && ./test_ppx_op.exe)
Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test_ppx/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/default/test_ppx && ./test_ppx_op_expected.exe)
Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test_ppx/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/test_print_style.expected", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/test_print_style.expected _build/default/test/test_print_style.exe.output
diff --git a/_build/default/test/test_print_style.expected b/_build/default/test/test_print_style.exe.output
index e51ab37..c55c8ad 100644
--- a/_build/default/test/test_print_style.expected
+++ b/_build/default/test/test_print_style.exe.output
@@ -1,5 +1,5 @@
-Welcome to OCANNL! Reading configuration defaults from /Users/lukstafi/ocannl/_build/default/test/ocannl_config.
+Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Testing print_style functionality:
File "test/einsum/test_conv_syntax.expected", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum/test_conv_syntax.expected _build/default/test/einsum/test_conv_syntax.exe.output
diff --git a/_build/default/test/einsum/test_conv_syntax.expected b/_build/default/test/einsum/test_conv_syntax.exe.output
index 4bd5a1c..0580e1e 100644
--- a/_build/default/test/einsum/test_conv_syntax.expected
+++ b/_build/default/test/einsum/test_conv_syntax.exe.output
@@ -1,5 +1,5 @@
-Welcome to OCANNL! Reading configuration defaults from /Users/lukstafi/ocannl/_build/default/test/einsum/ocannl_config.
+Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test/einsum/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Testing conv syntax parsing...
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition '' -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition 'Retrieving commandline, environment, or config file variable ocannl_log_level' -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition 'Found 0, in the config file' -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition 'Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/f0a1033dfc2d5d93ffe95903c3d73d99/default/test/ocannl_config.' -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/dune", lines 30-40, characters 0-281:
30 | (rule
31 | (alias runtest)
32 | (target
33 | (dir log_files))
34 | (action
35 | (run
36 | %{dep:micrograd_demo_logging.exe}
37 | "--ocannl_debug_backend=text"
38 | "--ocannl_log_file_stem=micrograd_demo_logging"
39 | "--ocannl_log_main_domain_to_stdout=false"
40 | "--ocannl_debug_log_to_stream_files=true")))
(cd _build/default/test && ./micrograd_demo_logging.exe --ocannl_debug_backend=text --ocannl_log_file_stem=micrograd_demo_logging --ocannl_log_main_domain_to_stdout=false --ocannl_debug_log_to_stream_files=true)
Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Retrieving commandline, environment, or config file variable ocannl_backend
Found multicore_cc, in the config file
Retrieving commandline, environment, or config file variable ocannl_cd_ident_style
Not found, using default heuristic
Retrieving commandline, environment, or config file variable ocannl_ll_ident_style
Not found, using default heuristic
Retrieving commandline, environment, or config file variable ocannl_prefer_backend_uniformity
Found true, in the config file
Retrieving commandline, environment, or config file variable ocannl_debug_log_to_stream_files
Found true, commandline --ocannl_debug_log_to_stream_files=true
Retrieving commandline, environment, or config file variable ocannl_cc_backend_optimization_level
Not found, using default 3
Retrieving commandline, environment, or config file variable ocannl_cc_backend_compiler_command
Not found, using default gcc
Retrieving commandline, environment, or config file variable ocannl_never_capture_stdout
Not found, using default false
Fatal error: exception File "src/printbox-text/PrintBox_text.ml", line 212, characters 6-12: Assertion failed
Raised at PrintBox_text.Output.Make_out.to_buf_aux_ in file "src/printbox-text/PrintBox_text.ml", line 212, characters 6-50
Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 19-42
Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41
Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41
Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41
Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41
Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41
Called from PrintBox_text.Output.Make_out.render in file "src/printbox-text/PrintBox_text.ml", line 242, characters 14-64
Called from PrintBox_text.output in file "src/printbox-text/PrintBox_text.ml", line 851, characters 2-31
Called from Minidebug_runtime.PrintBox.output_box in file "minidebug_runtime.ml", line 1527, characters 19-59
Called from Minidebug_runtime.PrintBox.close_log_impl.close_tree in file "minidebug_runtime.ml", line 1572, characters 6-38
Called from Backends.Add_buffer_retrieval_and_syncing.sync_routine.post in file "arrayjit/lib/backends.ml", lines 163-170, characters 13-15
Called from Ir__Task.run in file "arrayjit/lib/task.ml", line 17, characters 2-14
Re-raised at Ir__Task.run in file "arrayjit/lib/task.ml", lines 15-17, characters 20-14
Called from Dune__exe__Micrograd_demo_logging.(fun) in file "test/micrograd_demo_logging.ml", line 36, characters 2-16
Called from Utils.capture_stdout_logs in file "arrayjit/lib/utils.ml", line 825, characters 10-16
Re-raised at Utils.capture_stdout_logs in file "arrayjit/lib/utils.ml", line 856, characters 8-87
Called from Dune__exe__Micrograd_demo_logging in file "test/micrograd_demo_logging.ml", lines 35-39, characters 2-58
File "test/einsum/dune", line 13, characters 7-25:
13 | (name einsum_trivia_exec)
^^^^^^^^^^^^^^^^^^
(cd _build/default/test/einsum && ./einsum_trivia_exec.exe) > _build/default/test/einsum/einsum_trivia_exec.exe.output
Fatal error: exception Invalid_argument("Tnode.update_memory_mode: update 46 -> 35 inconsistent for r2x4x3")
Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45
Called from Ir__Low_level.visit_llc.(fun) in file "arrayjit/lib/low_level.ml", line 268, characters 13-53
Called from Base__Hashtbl.iteri in file "src/hashtbl.ml", line 281, characters 48-60
Re-raised at Base__Hashtbl.iteri in file "src/hashtbl.ml", line 288, characters 6-15
Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 875-876, characters 2-7
Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 869-886, characters 30-35
Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1149, characters 15-47
Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-75
Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-75
Called from Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 193-194, characters 4-45
Re-raised at Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 185-194, characters 34-47
Called from Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", line 356, characters 6-61
Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 355-356, characters 4-61
Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 354-362, characters 26-99
Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 494, characters 37-90
Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 488-497, characters 32-17
Called from Dune__exe__Einsum_trivia_exec in file "test/einsum/einsum_trivia_exec.ml", line 121, characters 12-47
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition primitive_ops.ml -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition zero2hero_1of7.ml -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition hello_world_op.ml -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/einsum/moons_demo_variant.expected", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum/moons_demo_variant.expected _build/default/test/einsum/moons_demo_variant.exe.output
diff --git a/_build/default/test/einsum/moons_demo_variant.expected b/_build/default/test/einsum/moons_demo_variant.exe.output
index 68171df..193d005 100644
--- a/_build/default/test/einsum/moons_demo_variant.expected
+++ b/_build/default/test/einsum/moons_demo_variant.exe.output
@@ -1,57 +1,52 @@
-Welcome to OCANNL! Reading configuration defaults from /Users/lukstafi/ocannl/_build/default/test/einsum/ocannl_config.
+Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test/einsum/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Half-moons scatterplot and decision boundary:
┌────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│****************************#***********************************************************************│
-│****************************###***###***#***********************************************************│
-│***************************#********#***#**********************************************************.│
-│********************#*********************##*#*#************************************************....│
-│******************#***#*****##*#*#*#****#*#*#*##**********************************************......│
-│**************#*#**###**#*##**##********#**#****###**#*************************************.........│
-│***********#***##***#**##*****************#*#*##*###*************************************...........│
-│***************#**###******************************##*#********************************.............│
-│*************#*****#******************************#*#***#*#*************************................│
-│*********#***#************************************##*###**#***********************..................│
-│******#***#*#*#******************************************#*#*******************.....................│
-│*****###****#*******************************************#********************.......................│
-│*******#*###***********************************************#*#************..........................│
-│**********##**********************%%*********************#**##**********............................│
-│*****#***#**********************%**%%********************************.............................%.│
-│**####*************************%**%**************************##*#**............................%...%│
-│****#**************************%%**%*************************#*#*..............................%....│
-│*#****##************************%*...%*********************##*..#...........................%...%%..│
-│****#*#***********************.%..%.%........................#..#.............................%.%...│
-│##*##*********************.......%.............................#.##.........................%.%.%%%.│
-│**#####*****************.......%...%%.......................................................%..%.%..│
-│**********************...........%.%.%%.....................#.#.#.#.............................%%..│
-│***##****************............%..%..%........................#.#.........................%...%...│
-│##**#**************................%...%.....................#..###......................%.%%.......│
-│###***************...............%.%.%...........................##............................%....│
-│**##************.................................................#......................%%.%..%.....│
-│**#************....................%%...........................##......................%%.%........│
-│**************........................%......................#..#...........................%.......│
-│************........................%%.%.......................##.....................%..%..........│
-│***********............................%%..............................................%%.%%........│
-│**********.............................%...%............................................%%%.........│
-│********...................................%..%....................................%.....%..........│
-│*******.................................%%..%.%................................%.%.%.%%.............│
-│******.....................................%%.%...%%.%.......................%%%....%.%.%...........│
-│*****....................................%.....%.%..%....................%.....%%.%.%...............│
-│***.........................................%%...%.%%%%..%.......%.........%%.%%....................│
-│**................................................%..%.......%..%%%%...%.%..%.%.....................│
-│*..................................................%.%.%............%..%...%........................│
-│..................................................%....%.%......%%...%.%............................│
-│..........................................................%..%......%%%.............................│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
+│****************************************************************************************************│
└────────────────────────────────────────────────────────────────────────────────────────────────────┘Loss:
┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 3.79e+1│- │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
+│ 2.92e+6│ - - │
│ │ │
│ │ │
│ │ │
@@ -61,124 +56,129 @@ Half-moons scatterplot and decision boundary:
│ │ │
│ │ │
│ │ │
+│ │ - - │
│ │ │
│ │ │
+│ │ - - │
+│ │ - - - - │
+│ │ - - │
+│ │ - - │
│ │ │
│l │ │
-│o │ │
-│s │ │
-│s │ │
-│ │ │
-│ │ │
+│o │ - - │
+│s │ - - │
+│s │ - - - - │
+│ │ - - │
+│ │ │
+│ │ - - │
+│ │ - - - - - - │
+│ │ - - │
+│ │ - - - - - - - - │
+│ │ - - - - │
+│ │ │
+│ │ │
+│ │ │
+│ │ - - - - │
+│ │ - - - - - - │
+│ │ - - │
+│ │ - - │
+│ │- - - - │
+│ │ - - - - - - - - │
+│ │ - - - - │
+│ 1.99e+5│ - - - - │
+├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
+│ │0.00 7.90e+1│
+│ │ step │
+└────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘Log-loss, for better visibility:
+┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
+│ 1.48e+1│ - - │
│ │ │
│ │ │
│ │ │
+│ │ - - │
+│ │ - - │
+│ │ - - - - │
+│ │ - - - - │
│ │ │
+│ │ - - │
+│ │ - - - - │
+│ │ - - - - │
│ │ │
+│ │ - - │
+│ │ - - - - - - - - │
+│ │ - - - - - - - - │
+│l │ - - - - │
+│o │ │
+│g │ │
│ │ │
-│ │ - │
+│l │ │
+│o │ - - - - │
+│s │ - - - - │
+│s │ - - - - │
│ │ │
+│ │ - - │
│ │ │
│ │ │
+│ │ - - │
+│ │- - │
+│ │ - - │
+│ │ - - - - │
+│ │ - - │
│ │ │
+│ │ - - - - │
│ │ │
│ │ │
+│ │ - - │
│ │ │
-│ │ - - - - │
-│ 0.00 │ - ---- --- --- ---- ---- ---- ---- -- - ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
+│ 1.22e+1│ - - │
├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ │0.00 7.90e+1│
│ │ step │
-└────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘Log-loss, for better visibility:
-┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 3.63 │- │
-│ │ │
-│ │ │
-│ │ │
-│ │ - │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ - │
-│ │ - │
-│ │ - - - │
-│ │ - - - - │
-│ │ -- - - - - - - - -- - - - - - -- │
-│ │ - - - - - - - -- - - -- - - - - │
-│ │ - - - │
-│l │ - - - -- - │
-│o │ - - │
-│g │ - - - - │
-│ │ - - - │
-│l │ - -- │
-│o │ - - - │
-│s │ - │
-│s │ - │
-│ │ - │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ │ │
-│ -1.00e+1│ - - - - - - - │
-├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
-│ │0.00 7.90e+1│
-│ │ step │
-└─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘
+└────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘
Learning rate:
-┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ -1.01e-1│ - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│l │ -- │
-│e │ - - │
-│a │ -- │
-│r │ - - │
-│n │ -- │
-│i │ - - │
-│n │ -- │
-│g │ - - │
-│ │ -- │
-│r │ - - │
-│a │ -- │
-│t │ - - │
-│e │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ │ -- │
-│ │ - - │
-│ -2.00e-1│--- │
-├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
-│ │0.00 7.90e+1│
-│ │ step │
-└─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘mlp_result's name: mlp_point
+┌──────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
+│ -8.24e-41│ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│l │ │
+│e │ │
+│a │ │
+│r │ │
+│n │ │
+│i │ │
+│n │ │
+│g │ │
+│ │ │
+│r │ │
+│a │ │
+│t │ │
+│e │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ │ │
+│ -8.24e-41│---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
+├──────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
+│ │0.00 7.90e+1│
+│ │ step │
+└──────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘mlp_result's name: mlp_point
(mlp moons_input) name: mlp_moons_input
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition einsum_trivia.ml -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition micrograd_demo.ml -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/dune", line 4, characters 7-30:
4 | (name moons_demo_parallel_run)
^^^^^^^^^^^^^^^^^^^^^^^
(cd _build/default/test && ./moons_demo_parallel_run.exe)
Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
("Set log_level to" 1)
└─{orphaned from #24}
Retrieving commandline, environment, or config file variable ocannl_backend
Found multicore_cc, in the config file
Properties of devices:
(multicore_devices
(device ((device_name CPU) (device_ordinal 0) (num_domains 72))))
@!Retrieving commandline, environment, or config file variable ocannl_prefer_backend_uniformity
Found true, in the config file
Retrieving commandline, environment, or config file variable ocannl_debug_log_to_stream_files
Not found, using default false
Retrieving commandline, environment, or config file variable ocannl_ll_ident_style
Not found, using default heuristic
Retrieving commandline, environment, or config file variable ocannl_cc_backend_optimization_level
Not found, using default 3
Retrieving commandline, environment, or config file variable ocannl_cc_backend_compiler_command
Not found, using default gcc
Fatal error: exception (Option.value_exn arrayjit/lib/backends.ml:289:28)
Raised at Base__Error.raise in file "src/error.ml" (inlined), line 9, characters 21-37
Called from Base__Option.value_exn in file "src/option.ml", line 119, characters 4-21
Called from Backends.Add_device.link_batch in file "arrayjit/lib/backends.ml", line 289, characters 5-44
Called from Backends.Raise_backend.link_batch in file "arrayjit/lib/backends.ml", line 467, characters 30-81
Called from Ocannl__Train.parallel_update.(fun) in file "lib/train.ml", line 270, characters 26-90
Re-raised at Ocannl__Train.parallel_update.(fun) in file "lib/train.ml", line 270, characters 10-91
Re-raised at Ocannl__Train.parallel_update.(fun) in file "lib/train.ml", lines 268-271, characters 23-18
Called from Base__Array0.mapi in file "src/array0.ml", line 142, characters 24-46
Called from Ocannl__Train.parallel_update in file "lib/train.ml", lines 268-271, characters 4-18
Re-raised at Ocannl__Train.parallel_update in file "lib/train.ml", lines 265-271, characters 2-18
Re-raised at Ocannl__Train.parallel_update in file "lib/train.ml", lines 233-318, characters 32-69
Called from Ocannl__Train.example_train_loop in file "lib/train.ml", lines 415-425, characters 4-39
Called from Dune__exe__Moons_demo_parallel_run.main in file "test/moons_demo_parallel_run.ml", lines 65-69, characters 4-8
Called from Base__Exn.protectx in file "src/exn.ml", line 79, characters 8-11
Re-raised at Base__Exn.raise_with_original_backtrace in file "src/exn.ml" (inlined), line 59, characters 2-50
Called from Base__Exn.protectx in file "src/exn.ml", line 86, characters 13-49
Called from Base__Exn.protect in file "src/exn.ml" (inlined), line 92, characters 26-49
Called from Dune__exe__Moons_demo_parallel_run in file "test/moons_demo_parallel_run.ml", line 90, characters 2-53
(cd _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition moons_demo_parallel.ml -source-tree-root .. -diff-cmd -)
Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/ocannl_config.
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/moons_demo_parallel.ml", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/moons_demo_parallel.ml _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/moons_demo_parallel.ml.corrected
diff --git a/_build/default/test/moons_demo_parallel.ml b/_build/.sandbox/206e79d926ef941d97a5bbc9a8bd5b0/default/test/moons_demo_parallel.ml.corrected
index ce09725..343032a 100644
--- a/_build/default/test/moons_demo_parallel.ml
+++ b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/moons_demo_parallel.ml.corrected
@@ -105,4 +105,24 @@ let%expect_test "Half-moons data parallel" =
]
in
PrintBox_text.output Stdio.stdout plot_loss);
- [%expect "Success"]
+ [%expect.unreachable]
+[@@expect.uncaught_exn {|
+ (* CR expect_test_collector: This test expectation appears to contain a backtrace.
+ This is strongly discouraged as backtraces are fragile.
+ Please change this test to not include a backtrace. *)
+ (Option.value_exn arrayjit/lib/backends.ml:289:28)
+ Raised at Base__Error.raise in file "src/error.ml" (inlined), line 9, characters 21-37
+ Called from Base__Option.value_exn in file "src/option.ml", line 119, characters 4-21
+ Called from Backends.Add_device.link_batch in file "arrayjit/lib/backends.ml", line 289, characters 5-44
+ Called from Backends.Raise_backend.link_batch in file "arrayjit/lib/backends.ml", line 467, characters 30-81
+ Called from Ocannl__Train.parallel_update.(fun) in file "lib/train.ml", line 270, characters 26-90
+ Re-raised at Ocannl__Train.parallel_update.(fun) in file "lib/train.ml", line 270, characters 10-91
+ Re-raised at Ocannl__Train.parallel_update.(fun) in file "lib/train.ml", lines 268-271, characters 23-18
+ Called from Base__Array0.mapi in file "src/array0.ml", line 142, characters 24-46
+ Called from Ocannl__Train.parallel_update in file "lib/train.ml", lines 268-271, characters 4-18
+ Re-raised at Ocannl__Train.parallel_update in file "lib/train.ml", lines 265-271, characters 2-18
+ Re-raised at Ocannl__Train.parallel_update in file "lib/train.ml", lines 233-318, characters 32-69
+ Called from Ocannl__Train.example_train_loop in file "lib/train.ml", lines 415-425, characters 4-39
+ Called from Tutorials__Moons_demo_parallel.(fun) in file "test/moons_demo_parallel.ml", lines 58-62, characters 4-8
+ Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28
+ |}]
File "test/zero2hero_1of7.ml", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/zero2hero_1of7.ml _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/zero2hero_1of7.ml.corrected
diff --git a/_build/default/test/zero2hero_1of7.ml b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/zero2hero_1of7.ml.corrected
index 0e9bd26..d2ca498 100644
--- a/_build/default/test/zero2hero_1of7.ml
+++ b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/zero2hero_1of7.ml.corrected
@@ -33,15 +33,15 @@ let%expect_test "Graph drawing recompile" =
6.00e+1
#16 grad_+_f_nd Virt/30
<void>
- #13 - Virt/152 │#2 5. Virt/40
+ #13 - Virt/40 │#2 5. Virt/40
<void> │<void>
#14 grad_- Virt/30 │
<void> │
- #11 *. Virt/152 │ #4 *. Virt/152 │
+ #11 *. Virt/40 │ #4 *. Virt/40 │
<void> │ <void> │
#12 grad_*. Virt/30 │ #5 grad_*. Virt/30 │
<void> │ <void> │
- #10 3. Virt/40│#7 **. Virt/152 │#3 4. Virt/40│#0 x │
+ #10 3. Virt/40│#7 **. Virt/40 │#3 4. Virt/40│#0 x │
<void> │<void> │<void> │ 5.00 │
│#8 grad_**. Virt/30│ │#1 grad_x Local/30│
│<void> │ │<void> │
@@ -96,7 +96,8 @@ let%expect_test "Graph drawing recompile" =
[%expect
{|
┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │ 1.00e+2│# │
+ │ 6.00e+1│ │
+ │ │ │
│ │ │
│ │ │
│ │ │
@@ -109,7 +110,6 @@ let%expect_test "Graph drawing recompile" =
│ │ │
│ │ │
│ │ │
- │ │ # │
│ │ │
│ │ │
│ │ │
@@ -119,23 +119,23 @@ let%expect_test "Graph drawing recompile" =
│x │ │
│) │ │
│ │ │
- │ │ # │
│ │ │
│ │ │
- │ │ #│
│ │ │
│ │ │
│ │ │
│ │ │
- │ │ # │
│ │ │
- │ │ # │
│ │ │
│ │ │
- │ │ # │
- │ │ # │
│ │ │
- │ 4.00 │ # # │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ 6.00e+1│# # # # # # # # # #│
├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ │-5.00 4.00│
│ │ x │
@@ -276,15 +276,15 @@ let%expect_test "Simple gradients hosted" =
<not-in-yet>
#13 grad_*._l Host&stream/41
<not-in-yet>
- #8 +_d Host&stream/41 │#10 f Host&shared/39
+ #8 +_d Host&stream/41 │#10 f Host&stream/41
<not-in-yet> │<not-in-yet>
#9 grad_+_d Host&stream/41 │#11 grad_f Host&stream/41
<not-in-yet> │<not-in-yet>
- #4 *._e Host&stream/41 │#6 c Host&shared/39 │
+ #4 *._e Host&stream/41 │#6 c Host&stream/41 │
<not-in-yet> │<not-in-yet> │
#5 grad_*._e Host&stream/41 │#7 grad_c Host&stream/41│
<not-in-yet> │<not-in-yet> │
- #0 a Host&shared/39 │#2 b Host&shared/39 │ │
+ #0 a Host&stream/41 │#2 b Host&stream/41 │ │
<not-in-yet> │<not-in-yet> │ │
#1 grad_a Host&stream/41│#3 grad_b Host&stream/41│ │
<not-in-yet> │<not-in-yet> │ │
@@ -324,15 +324,15 @@ let%expect_test "Simple gradients hosted" =
#13 grad_*._l
1.00
#8 +_d │#10 f
- 4.00 │ -2.40
+ 4.00 │ -2.00
#9 grad_+_d │#11 grad_f
-2.00 │ 4.00
#4 *._e │#6 c │
- -6.00 │ 1.02e+1 │
+ -6.00 │ 1.00e+1 │
#5 grad_*._e │#7 grad_c│
-2.00 │ -2.00 │
#0 a │#2 b │ │
- 1.40 │ -2.60 │ │
+ 2.00 │ -3.00 │ │
#1 grad_a│#3 grad_b│ │
6.00 │ -4.00 │ │
|}];
@@ -344,21 +344,21 @@ let%expect_test "Simple gradients hosted" =
[%expect
{|
#12 *._l
- -1.57e+1
+ -8.00
#13 grad_*._l
1.00
#8 +_d │#10 f
- 6.56 │ -2.40
+ 4.00 │ -2.00
#9 grad_+_d │#11 grad_f
- -2.40 │ 6.56
+ -2.00 │ 4.00
#4 *._e │#6 c │
- -3.64 │ 1.02e+1 │
+ -6.00 │ 1.00e+1 │
#5 grad_*._e │#7 grad_c│
- -2.40 │ -2.40 │
+ -2.00 │ -2.00 │
#0 a │#2 b │ │
- 1.40 │ -2.60 │ │
+ 2.00 │ -3.00 │ │
#1 grad_a│#3 grad_b│ │
- 6.24 │ -3.36 │ │
+ 6.00 │ -4.00 │ │
|}]
let%expect_test "Simple gradients virtual" =
@@ -386,15 +386,15 @@ let%expect_test "Simple gradients virtual" =
<not-in-yet>
#8 +_d unknown │#10 f Host-non-const/24
<not-in-yet> │<not-in-yet>
- #9 grad_+_d unknown │#11 grad_f Material/28
+ #9 grad_+_d unknown │#11 grad_f Non-virt/26
<not-in-yet> │<not-in-yet>
#4 *._e unknown │#6 c Host-non-const/24│
<not-in-yet> │<not-in-yet> │
- #5 grad_*._e unknown │#7 grad_c Material/28 │
+ #5 grad_*._e unknown │#7 grad_c Non-virt/26 │
<not-in-yet> │<not-in-yet> │
#0 a Host-non-const/24│#2 b Host-non-const/24│ │
<not-in-yet> │<not-in-yet> │ │
- #1 grad_a Material/28 │#3 grad_b Material/28 │ │
+ #1 grad_a Non-virt/26 │#3 grad_b Non-virt/26 │ │
<not-in-yet> │<not-in-yet> │ │
|}];
let grad_routine = Train.to_routine (module Backend) ctx IDX.empty grad in
@@ -402,22 +402,22 @@ let%expect_test "Simple gradients virtual" =
Tensor.print_tree ~spy:true ~with_grad:true ~depth:9 l;
[%expect
{|
- #12 *._l Host&stream/41
- <not-in-yet>
- #13 grad_*._l Virt/40
- <not-in-yet>
- #8 +_d Local/46 │#10 f Host&shared/39
- <not-in-yet> │<not-in-yet>
- #9 grad_+_d Virt/40 │#11 grad_f Dev-stream/41
- <not-in-yet> │<not-in-yet>
- #4 *._e Virt/152 │#6 c Host&shared/39 │
- <not-in-yet> │<not-in-yet> │
- #5 grad_*._e Virt/40 │#7 grad_c Dev-stream/41│
- <not-in-yet> │<not-in-yet> │
- #0 a Host&shared/39 │#2 b Host&shared/39 │ │
- <not-in-yet> │<not-in-yet> │ │
- #1 grad_a Dev-stream/41│#3 grad_b Dev-stream/41│ │
- <not-in-yet> │<not-in-yet> │ │
+ #12 *._l Host&stream/41
+ <not-in-yet>
+ #13 grad_*._l Virt/40
+ <not-in-yet>
+ #8 +_d Virt/40 │#10 f Host&stream/41
+ <not-in-yet> │<not-in-yet>
+ #9 grad_+_d Virt/40 │#11 grad_f Local/46
+ <not-in-yet> │<not-in-yet>
+ #4 *._e Virt/40 │#6 c Host&stream/41│
+ <not-in-yet> │<not-in-yet> │
+ #5 grad_*._e Virt/40 │#7 grad_c Local/46 │
+ <not-in-yet> │<not-in-yet> │
+ #0 a Host&stream/41│#2 b Host&stream/41│ │
+ <not-in-yet> │<not-in-yet> │ │
+ #1 grad_a Local/46 │#3 grad_b Local/46 │ │
+ <not-in-yet> │<not-in-yet> │ │
|}];
(* Do not update the params: all values and gradients will be at initial points, which are
specified in the tensor in the brackets. *)
@@ -425,22 +425,22 @@ let%expect_test "Simple gradients virtual" =
Tensor.print_tree ~with_grad:true ~depth:9 l;
[%expect
{|
- #12 *._l
- -8.00
- #13 grad_*._l Virt/40
- <void>
- #8 +_d Local/46 │#10 f
- <void> │ -2.00
- #9 grad_+_d Virt/40 │#11 grad_f Dev-stream/41
- <void> │<void>
- #4 *._e Virt/152 │#6 c │
- <void> │ 1.00e+1 │
- #5 grad_*._e Virt/40 │#7 grad_c Dev-stream/41│
- <void> │<void> │
- #0 a │#2 b │ │
- 2.00 │ -3.00 │ │
- #1 grad_a Dev-stream/41│#3 grad_b Dev-stream/41│ │
- <void> │<void> │ │
+ #12 *._l
+ -8.00
+ #13 grad_*._l Virt/40
+ <void>
+ #8 +_d Virt/40 │#10 f
+ <void> │ -2.00
+ #9 grad_+_d Virt/40 │#11 grad_f Local/46
+ <void> │<void>
+ #4 *._e Virt/40 │#6 c │
+ <void> │ 1.00e+1 │
+ #5 grad_*._e Virt/40 │#7 grad_c Local/46│
+ <void> │<void> │
+ #0 a │#2 b │ │
+ 2.00 │ -3.00 │ │
+ #1 grad_a Local/46│#3 grad_b Local/46│ │
+ <void> │<void> │ │
|}];
(* Only now compile the SGD update. *)
let sgd_routine = Train.to_routine (module Backend) grad_routine.context IDX.empty sgd in
@@ -451,22 +451,22 @@ let%expect_test "Simple gradients virtual" =
Tensor.print_tree ~with_grad:true ~depth:9 l;
[%expect
{|
- #12 *._l
- -8.00
- #13 grad_*._l Virt/40
- <void>
- #8 +_d Local/46 │#10 f
- <void> │ -2.40
- #9 grad_+_d Virt/40 │#11 grad_f Dev-stream/41
- <void> │<void>
- #4 *._e Virt/152 │#6 c │
- <void> │ 1.02e+1 │
- #5 grad_*._e Virt/40 │#7 grad_c Dev-stream/41│
- <void> │<void> │
- #0 a │#2 b │ │
- 1.40 │ -2.60 │ │
- #1 grad_a Dev-stream/41│#3 grad_b Dev-stream/41│ │
- <void> │<void> │ │
+ #12 *._l
+ -8.00
+ #13 grad_*._l Virt/40
+ <void>
+ #8 +_d Virt/40 │#10 f
+ <void> │ -2.00
+ #9 grad_+_d Virt/40 │#11 grad_f Local/46
+ <void> │<void>
+ #4 *._e Virt/40 │#6 c │
+ <void> │ 1.00e+1 │
+ #5 grad_*._e Virt/40 │#7 grad_c Local/46│
+ <void> │<void> │
+ #0 a │#2 b │ │
+ 2.00 │ -3.00 │ │
+ #1 grad_a Local/46│#3 grad_b Local/46│ │
+ <void> │<void> │ │
|}];
(* Now the params will remain as above, but both param gradients and the values and gradients of
other nodes will change thanks to the forward and backward passes. *)
@@ -474,22 +474,22 @@ let%expect_test "Simple gradients virtual" =
Tensor.print_tree ~with_grad:true ~depth:9 l;
[%expect
{|
- #12 *._l
- -1.57e+1
- #13 grad_*._l Virt/40
- <void>
- #8 +_d Local/46 │#10 f
- <void> │ -2.40
- #9 grad_+_d Virt/40 │#11 grad_f Dev-stream/41
- <void> │<void>
- #4 *._e Virt/152 │#6 c │
- <void> │ 1.02e+1 │
- #5 grad_*._e Virt/40 │#7 grad_c Dev-stream/41│
- <void> │<void> │
- #0 a │#2 b │ │
- 1.40 │ -2.60 │ │
- #1 grad_a Dev-stream/41│#3 grad_b Dev-stream/41│ │
- <void> │<void> │ │
+ #12 *._l
+ -8.00
+ #13 grad_*._l Virt/40
+ <void>
+ #8 +_d Virt/40 │#10 f
+ <void> │ -2.00
+ #9 grad_+_d Virt/40 │#11 grad_f Local/46
+ <void> │<void>
+ #4 *._e Virt/40 │#6 c │
+ <void> │ 1.00e+1 │
+ #5 grad_*._e Virt/40 │#7 grad_c Local/46│
+ <void> │<void> │
+ #0 a │#2 b │ │
+ 2.00 │ -3.00 │ │
+ #1 grad_a Local/46│#3 grad_b Local/46│ │
+ <void> │<void> │ │
|}]
let%expect_test "tanh plot" =
File "test/micrograd_demo.ml", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/micrograd_demo.ml _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/micrograd_demo.ml.corrected
diff --git a/_build/default/test/micrograd_demo.ml b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/micrograd_demo.ml.corrected
index 59a566d..b11841d 100644
--- a/_build/default/test/micrograd_demo.ml
+++ b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/micrograd_demo.ml.corrected
@@ -184,46 +184,46 @@ let%expect_test "Micrograd half-moons example" =
{|
Half-moons scatterplot and decision boundary:
┌────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │****************************#***********************************************************************│
- │****************************###***###***#***********************************************************│
- │***************************#********#***#***********************************************************│
- │********************#*********************##*#*#****************************************************│
- │******************#***#*****##*#*#*#****#*#*#*##****************************************************│
- │**************#*#**###**#*##**##********#**#****###**#**********************************************│
- │***********#***##***#**##*****************#*#*##*###**********************************************..│
- │***************#**###******************************##*#******************************************...│
- │*************#*****#******************************#*#***#*#*************************************....│
- │*********#***#******************......************##*###**#************************************.....│
- │******#***#*#*#****************........******************#*#*********************************.......│
- │*****###****#****************...........****************#***********************************........│
- │*******#*###****************.............******************#*#*****************************.........│
- │**********##***************.......%%......***************#**##****************************..........│
- │*****#***#****************......%..%%......*********************************************..........%.│
- │**####*******************......%..%..........****************##*#**********************........%...%│
- │****#*******************.......%%..%..........***************#*#**********************.........%....│
- │*#****##***************.........%....%.........************##***#********************.......%...%%..│
- │****#*#***************.........%..%.%...........*************#**#*******************..........%.%...│
- │##*##****************............%...............**************#*##****************.........%.%.%%%.│
- │**#####************............%...%%.............*******************************...........%..%.%..│
- │******************...............%.%.%%............*********#*#*#*#*************................%%..│
- │***##************................%..%..%............************#*#************.............%...%...│
- │##**#***********...................%...%.............********#**###***********...........%.%%.......│
- │###************..................%.%.%................***********##**********..................%....│
- │**##**********.........................................**********#*********.............%%.%..%.....│
- │**#**********......................%%...................********##********..............%%.%........│
- │************..........................%..................****#**#********...................%.......│
- │***********.........................%%.%..................*****##******...............%..%..........│
- │**********.............................%%..................***********.................%%.%%........│
- │*********..............................%...%................*******.....................%%%.........│
- │********...................................%..%....................................%.....%..........│
- │*******.................................%%..%.%................................%.%.%.%%.............│
- │******.....................................%%.%...%%.%.......................%%%....%.%.%...........│
- │*****....................................%.....%.%..%....................%.....%%.%.%...............│
- │****........................................%%...%.%%%%..%.......%.........%%.%%....................│
- │***...............................................%..%.......%..%%%%...%.%..%.%.....................│
- │**.................................................%.%.%............%..%...%........................│
- │*.................................................%....%.%......%%...%.%............................│
- │..........................................................%..%......%%%.............................│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
+ │****************************************************************************************************│
└────────────────────────────────────────────────────────────────────────────────────────────────────┘
|}];
Stdio.printf "Loss:\n%!";
@@ -236,156 +236,156 @@ let%expect_test "Micrograd half-moons example" =
{|
Loss:
┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │ 3.79e+1│- │
+ │ 5.79e+6│ - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
│ │ │
│ │ │
│ │ │
│ │ │
+ │ │ - - - - - - - - - -│
│ │ │
+ │ │ - - - - - - - - - - - - - - - - - - - -│
+ │ │ - - - - - - - - - - - - - - - - - - - - │
│ │ │
│ │ │
│ │ │
│ │ │
+ │ │ - - - - - - - - - - - - - - - - - - - -│
+ │ │- - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - -│
+ │l │ - - - - - - - - - - - - - - - - - - - - │
+ │o │ - - - - - - - - - - │
+ │s │ - - - - - - - - - - - - - - - - - - - - │
+ │s │ - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - │
│ │ │
+ │ │- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
+ │ │- - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
│ │ │
+ │ │ - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - │
│ │ │
+ │ │- - - - - - - - - - │
│ │ │
│ │ │
│ │ │
- │ │ │
- │ │ │
- │l │ │
- │o │ │
- │s │ │
+ │ 1.69e+6│ - - - - - - - - - - │
+ ├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
+ │ │0.00 3.99e+2│
+ │ │ step │
+ └────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘
+ |}];
+ Stdio.printf "Log-loss, for better visibility:\n%!";
+ let plot_loss =
+ PrintBox_utils.plot ~x_label:"step" ~y_label:"log loss"
+ [ Line_plot { points = Array.of_list_rev !log_losses; content = PrintBox.line "-" } ]
+ in
+ PrintBox_text.output Stdio.stdout plot_loss;
+ [%expect
+ {|
+ Log-loss, for better visibility:
+ ┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
+ │ 1.55e+1│ - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
+ │ │ │
+ │ │ │
+ │ │ - - - - - - - - - -│
+ │ │ - - - - - - - - - - - - - - - - - - - -│
+ │ │ - - - - - - - - - - - - - - - - - - - - │
+ │ │ │
+ │ │ │
+ │ │ - - - - - - - - - - - - - - - - - - - -│
+ │ │- - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - -│
+ │ │ - - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - │
+ │l │ - - - - - - - - - - │
+ │o │ - - - - - - - - - - │
+ │g │- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
+ │l │- - - - - - - - - - │
+ │o │ - - - - - - - - - - │
+ │s │ - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - │
│s │ │
│ │ │
+ │ │ - - - - - - - - - - │
│ │ │
+ │ │ - - - - - - - - - - - - - - - - - - - - │
+ │ │ - - - - - - - - - - │
+ │ │ - - - - - - - - - - - - - - - - - - - - │
│ │ │
│ │ │
+ │ │- - - - - - - - - - │
│ │ │
│ │ │
│ │ │
│ │ │
- │ │- │
- │ │ │
- │ │ │
│ │ │
│ │ │
- │ │ │
- │ │ │
- │ │ │
- │ │- - - │
- │ 0.00 │----------------------------------------------------------------------------------------------------│
+ │ 1.43e+1│ - - - - - - - - - - │
├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ │0.00 3.99e+2│
│ │ step │
└────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘
|}];
- Stdio.printf "Log-loss, for better visibility:\n%!";
- let plot_loss =
- PrintBox_utils.plot ~x_label:"step" ~y_label:"log loss"
- [ Line_plot { points = Array.of_list_rev !log_losses; content = PrintBox.line "-" } ]
+ Stdio.printf "\nLearning rate:\n%!";
+ let plot_lr =
+ PrintBox_utils.plot ~x_label:"step" ~y_label:"learning rate"
+ [ Line_plot { points = Array.of_list_rev !learning_rates; content = PrintBox.line "-" } ]
in
- PrintBox_text.output Stdio.stdout plot_loss;
+ PrintBox_text.output Stdio.stdout plot_lr;
[%expect
{|
- Log-loss, for better visibility:
+ Learning rate:
┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │ 3.63 │- │
+ │ 6.41e+27│ │
│ │ │
│ │ │
│ │ │
- │ │- │
│ │ │
│ │ │
│ │ │
│ │ │
- │ │- │
- │ │ - - │
- │ │ - - - │
- │ │------ - │
- │ │ ---- - - --- - - - - - │
- │ │ -- - - - --- -- - -- - │
- │ │ - -- - - --- - - --- ----- - -- │
- │l │ -- - - - - --- - - - │
- │o │ - - - -- - -- - - - - - │
- │g │ - - - - -- - - - - - - - │
- │ │ - - - -- - - - -- - - - │
- │l │ - - -- -- - - - - -│
- │o │ - - - - - - - - - - - │
- │s │ - - - - -- - │
- │s │ - - │
- │ │ - - - - - - - - - │
- │ │ - - - - - │
- │ │ - - │
- │ │ - - - │
- │ │ - - │
│ │ │
- │ │ - │
- │ │ - │
│ │ │
│ │ │
│ │ │
│ │ │
- │ │ - │
+ │l │ │
+ │e │ │
+ │a │ │
+ │r │ │
+ │n │ │
+ │i │ │
+ │n │ │
+ │g │ │
│ │ │
+ │r │ │
+ │a │ │
+ │t │ │
+ │e │ │
│ │ │
- │ -1.00e+1│ - - - - ----- -- --------------- ---------------------------------------------------------│
- ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
- │ │0.00 3.99e+2│
- │ │ step │
- └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
- Stdio.printf "\nLearning rate:\n%!";
- let plot_lr =
- PrintBox_utils.plot ~x_label:"step" ~y_label:"learning rate"
- [ Line_plot { points = Array.of_list_rev !learning_rates; content = PrintBox.line "-" } ]
- in
- PrintBox_text.output Stdio.stdout plot_lr;
- [%expect
- {|
- Learning rate:
- ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │ -1.00e-1│ -│
- │ │ ---│
- │ │ ---- │
- │ │ --- │
- │ │ ---- │
- │ │ --- │
- │ │ --- │
- │ │ --- │
- │ │ --- │
- │ │ ---- │
- │ │ --- │
- │ │ ---- │
- │ │ --- │
- │l │ ---- │
- │e │ --- │
- │a │ --- │
- │r │ --- │
- │n │ --- │
- │i │ ---- │
- │n │ --- │
- │g │ ---- │
- │ │ --- │
- │r │ ---- │
- │a │ --- │
- │t │ --- │
- │e │ --- │
- │ │ --- │
- │ │ ---- │
- │ │ --- │
- │ │ ---- │
- │ │ --- │
- │ │ ---- │
- │ │ --- │
- │ │ --- │
- │ │ --- │
- │ │ --- │
- │ │ ---- │
- │ │ --- │
- │ │ ---- │
- │ -2.00e-1│--- │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ │ │
+ │ 6.41e+27│----------------------------------------------------------------------------------------------------│
├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ │0.00 3.99e+2│
│ │ step │
File "test/hello_world_op.ml", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/hello_world_op.ml _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/hello_world_op.ml.corrected
diff --git a/_build/default/test/hello_world_op.ml b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/hello_world_op.ml.corrected
index 0cea005..c56f85c 100644
--- a/_build/default/test/hello_world_op.ml
+++ b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/hello_world_op.ml.corrected
@@ -102,49 +102,15 @@ let%expect_test "Print constant tensor" =
let%op hey = [ (1, 2, 3); (4, 5, 6) ] in
Train.forward_and_forget backend ctx hey;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ hey;
- [%expect
- {|
- [0]: [ 1.00 , 2.00 , 3.00 ; 4.00 , 5.00 , 6.00 ]_hey shape 1:3->0:2 [
- 1.00 , 2.00 , 3.00
- ; 4.00 , 5.00 , 6.00
- ]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ hey;
- [%expect
- {|
- ┌────────────────────────────────────────────────────────────────────────┐
- │[0]: [ 1.00 , 2.00 , 3.00 ; 4.00 , 5.00 , 6.00 ]_hey shape 1:3->0:2 │
- │┌──────┬──────────────────┐ │
- ││ │axis 1 │ │
- │├──────┼──────────────────┤ │
- ││axis 0│ 1.00 2.00 3.00 │ │
- ││ │ 4.00 5.00 6.00 │ │
- │└──────┴──────────────────┘ │
- └────────────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let%op hoo = [| [ 1; 2; 3 ]; [ 4; 5; 6 ] |] in
Train.forward_and_forget backend ctx hoo;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ hoo;
- [%expect
- {|
- [1]: [| [ 1.00 ; 2.00 ; 3.00 ] ; [ 4.00 ; 5.00 ; 6.00 ] |]_hoo shape 0:2|1:3 [|
- [ 1.00 ; 2.00 ; 3.00 ]
- ; [ 4.00 ; 5.00 ; 6.00 ]
- |]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ hoo;
- [%expect
- {|
- ┌─────────────────────────────────────────────────────────────────────────────┐
- │[1]: [| [ 1.00 ; 2.00 ; 3.00 ] ; [ 4.00 ; 5.00 ; 6.00 ] |]_hoo shape 0:2|1:3 │
- │┌──────┬──────────────────┐ │
- ││ │axis 1 │ │
- │├──────┼──────────────────┤ │
- ││axis 0│ 1.00 2.00 3.00 │ │
- ││ │ 4.00 5.00 6.00 │ │
- │└──────┴──────────────────┘ │
- └─────────────────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let%op hey2 =
[
((1, 2, 3), (4, 5, 6));
@@ -155,31 +121,9 @@ let%expect_test "Print constant tensor" =
in
Train.forward_and_forget backend ctx hey2;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ hey2;
- [%expect
- {|
- [2]: c4x2x3_hey2 shape 1:2,2:3->0:4 [
- ( 1.00 , 2.00 , 3.00 ) , ( 4.00 , 5.00 , 6.00 )
- ; ( 7.00 , 8.00 , 9.00 ) , ( 10.00 , 11.00 , 12.00 )
- ; ( 13.00 , 14.00 , 15.00 ) , ( 16.00 , 17.00 , 18.00 )
- ; ( 19.00 , 20.00 , 21.00 ) , ( 22.00 , 23.00 , 24.00 )
- ]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ hey2;
- [%expect
- {|
- ┌────────────────────────────────────────────────────────────────┐
- │[2]: c4x2x3_hey2 shape 1:2,2:3->0:4 │
- │┌──────┬───────────────────────────┬───────────────────────────┐│
- ││ │0 @ 1 │1 @ 1 ││
- ││ │axis 2 │axis 2 ││
- │├──────┼───────────────────────────┼───────────────────────────┤│
- ││axis 0│ 1.00 2.00 3.00 │ 4.00 5.00 6.00 ││
- ││ │ 7.00 8.00 9.00 │ 1.00e+1 1.10e+1 1.20e+1 ││
- ││ │ 1.30e+1 1.40e+1 1.50e+1 │ 1.60e+1 1.70e+1 1.80e+1 ││
- ││ │ 1.90e+1 2.00e+1 2.10e+1 │ 2.20e+1 2.30e+1 2.40e+1 ││
- │└──────┴───────────────────────────┴───────────────────────────┘│
- └────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let%op hoo2 =
[|
[ [ 1; 2; 3 ]; [ 4; 5; 6 ] ];
@@ -190,29 +134,9 @@ let%expect_test "Print constant tensor" =
in
Train.forward_and_forget backend ctx hoo2;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ hoo2;
- [%expect
- {|
- [3]: c4x2x3_hoo2 shape 0:4|1:2,2:3 [|
- [ [ 1.00 ; 2.00 ; 3.00 ] ; [ 4.00 ; 5.00 ; 6.00 ] ]
- ; [ [ 7.00 ; 8.00 ; 9.00 ] ; [ 10.00 ; 11.00 ; 12.00 ] ]
- ; [ [ 13.00 ; 14.00 ; 15.00 ] ; [ 16.00 ; 17.00 ; 18.00 ] ]
- ; [ [ 19.00 ; 20.00 ; 21.00 ] ; [ 22.00 ; 23.00 ; 24.00 ] ]
- |]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ hoo2;
- [%expect
- {|
- ┌───────────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[3]: c4x2x3_hoo2 shape 0:4|1:2,2:3 │
- │┌──────┬──────────────────┬───────────────────────────┬───────────────────────────┬───────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 │2 @ 0 │3 @ 0 ││
- ││ │axis 2 │axis 2 │axis 2 │axis 2 ││
- │├──────┼──────────────────┼───────────────────────────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 1.00 2.00 3.00 │ 7.00 8.00 9.00 │ 1.30e+1 1.40e+1 1.50e+1 │ 1.90e+1 2.00e+1 2.10e+1 ││
- ││ │ 4.00 5.00 6.00 │ 1.00e+1 1.10e+1 1.20e+1 │ 1.60e+1 1.70e+1 1.80e+1 │ 2.20e+1 2.30e+1 2.40e+1 ││
- │└──────┴──────────────────┴───────────────────────────┴───────────────────────────┴───────────────────────────┘│
- └───────────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let%op heyhoo =
[|
[| [ 1; 2; 3 ]; [ 4; 5; 6 ] |];
@@ -223,29 +147,9 @@ let%expect_test "Print constant tensor" =
in
Train.forward_and_forget backend ctx heyhoo;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ heyhoo;
- [%expect
- {|
- [4]: c4x2x3_heyhoo shape 0:4,1:2|2:3 [|
- [| [ 1.00 ; 2.00 ; 3.00 ] ; [ 4.00 ; 5.00 ; 6.00 ] |]
- ; [| [ 7.00 ; 8.00 ; 9.00 ] ; [ 10.00 ; 11.00 ; 12.00 ] |]
- ; [| [ 13.00 ; 14.00 ; 15.00 ] ; [ 16.00 ; 17.00 ; 18.00 ] |]
- ; [| [ 19.00 ; 20.00 ; 21.00 ] ; [ 22.00 ; 23.00 ; 24.00 ] |]
- |]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ heyhoo;
- [%expect
- {|
- ┌───────────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[4]: c4x2x3_heyhoo shape 0:4,1:2|2:3 │
- │┌──────┬──────────────────┬───────────────────────────┬───────────────────────────┬───────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 │2 @ 0 │3 @ 0 ││
- ││ │axis 2 │axis 2 │axis 2 │axis 2 ││
- │├──────┼──────────────────┼───────────────────────────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 1.00 2.00 3.00 │ 7.00 8.00 9.00 │ 1.30e+1 1.40e+1 1.50e+1 │ 1.90e+1 2.00e+1 2.10e+1 ││
- ││ │ 4.00 5.00 6.00 │ 1.00e+1 1.10e+1 1.20e+1 │ 1.60e+1 1.70e+1 1.80e+1 │ 2.20e+1 2.30e+1 2.40e+1 ││
- │└──────┴──────────────────┴───────────────────────────┴───────────────────────────┴───────────────────────────┘│
- └───────────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let%op heyhoo2 =
[|
[| [ [ 1; 31 ]; [ 2; 32 ]; [ 3; 33 ] ]; [ [ 4; 34 ]; [ 5; 35 ]; [ 6; 36 ] ] |];
@@ -256,54 +160,9 @@ let%expect_test "Print constant tensor" =
in
Train.forward_and_forget backend ctx heyhoo2;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ heyhoo2;
- [%expect
- {|
- [5]: c4x2x3x2_heyhoo2 shape 0:4,1:2|2:3,3:2 [|
- [|
- [ [ 1.00 ; 31.00 ] ; [ 2.00 ; 32.00 ] ; [ 3.00 ; 33.00 ] ]
- ; [ [ 4.00 ; 34.00 ] ; [ 5.00 ; 35.00 ] ; [ 6.00 ; 36.00 ] ]
- |]
- ; [|
- [ [ 7.00 ; 37.00 ] ; [ 8.00 ; 38.00 ] ; [ 9.00 ; 39.00 ] ]
- ; [ [ 10.00 ; 40.00 ] ; [ 11.00 ; 41.00 ] ; [ 12.00 ; 42.00 ] ]
- |]
- ; [|
- [ [ 13.00 ; 43.00 ] ; [ 14.00 ; 44.00 ] ; [ 15.00 ; 45.00 ] ]
- ; [ [ 16.00 ; 46.00 ] ; [ 17.00 ; 47.00 ] ; [ 18.00 ; 48.00 ] ]
- |]
- ; [|
- [ [ 19.00 ; 49.00 ] ; [ 20.00 ; 50.00 ] ; [ 21.00 ; 51.00 ] ]
- ; [ [ 22.00 ; 52.00 ] ; [ 23.00 ; 53.00 ] ; [ 24.00 ; 54.00 ] ]
- |]
- |]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ heyhoo2;
- [%expect
- {|
- ┌──────────────────────────────────────────────┐
- │[5]: c4x2x3x2_heyhoo2 shape 0:4,1:2|2:3,3:2 │
- │┌──────┬──────────────────┬──────────────────┐│
- ││ │0 @ 1 │1 @ 1 ││
- ││ │axis 3 │axis 3 ││
- │├──────┼──────────────────┼──────────────────┤│
- ││0 @ 0 │ 1.00 3.10e+1 │ 4.00 3.40e+1 ││
- ││axis 2│ 2.00 3.20e+1 │ 5.00 3.50e+1 ││
- ││ │ 3.00 3.30e+1 │ 6.00 3.60e+1 ││
- │├──────┼──────────────────┼──────────────────┤│
- ││1 @ 0 │ 7.00 3.70e+1 │ 1.00e+1 4.00e+1 ││
- ││axis 2│ 8.00 3.80e+1 │ 1.10e+1 4.10e+1 ││
- ││ │ 9.00 3.90e+1 │ 1.20e+1 4.20e+1 ││
- │├──────┼──────────────────┼──────────────────┤│
- ││2 @ 0 │ 1.30e+1 4.30e+1 │ 1.60e+1 4.60e+1 ││
- ││axis 2│ 1.40e+1 4.40e+1 │ 1.70e+1 4.70e+1 ││
- ││ │ 1.50e+1 4.50e+1 │ 1.80e+1 4.80e+1 ││
- │├──────┼──────────────────┼──────────────────┤│
- ││3 @ 0 │ 1.90e+1 4.90e+1 │ 2.20e+1 5.20e+1 ││
- ││axis 2│ 2.00e+1 5.00e+1 │ 2.30e+1 5.30e+1 ││
- ││ │ 2.10e+1 5.10e+1 │ 2.40e+1 5.40e+1 ││
- │└──────┴──────────────────┴──────────────────┘│
- └──────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let%op heyhoo3 =
[|
[|
@@ -318,63 +177,9 @@ let%expect_test "Print constant tensor" =
in
Train.forward_and_forget backend ctx heyhoo3;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ heyhoo3;
- [%expect
- {|
- [6]: c2x2x2x3x2_heyhoo3 shape 0:2,1:2|2:2,3:3,4:2 [|
- [|
- [
- [ [ 1.00 ; 31.00 ] ; [ 2.00 ; 32.00 ] ; [ 3.00 ; 33.00 ] ]
- ; [ [ 4.00 ; 34.00 ] ; [ 5.00 ; 35.00 ] ; [ 6.00 ; 36.00 ] ]
- ]
- ; [
- [ [ 7.00 ; 37.00 ] ; [ 8.00 ; 38.00 ] ; [ 9.00 ; 39.00 ] ]
- ; [ [ 10.00 ; 40.00 ] ; [ 11.00 ; 41.00 ] ; [ 12.00 ; 42.00 ] ]
- ]
- |]
- ; [|
- [
- [ [ 13.00 ; 43.00 ] ; [ 14.00 ; 44.00 ] ; [ 15.00 ; 45.00 ] ]
- ; [ [ 16.00 ; 46.00 ] ; [ 17.00 ; 47.00 ] ; [ 18.00 ; 48.00 ] ]
- ]
- ; [
- [ [ 19.00 ; 49.00 ] ; [ 20.00 ; 50.00 ] ; [ 21.00 ; 51.00 ] ]
- ; [ [ 22.00 ; 52.00 ] ; [ 23.00 ; 53.00 ] ; [ 24.00 ; 54.00 ] ]
- ]
- |]
- |]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ heyhoo3;
- [%expect
- {|
- ┌──────────────────────────────────────────────────┐
- │[6]: c2x2x2x3x2_heyhoo3 shape 0:2,1:2|2:2,3:3,4:2 │
- │┌──────┬───────────────┬──────────────────┐ │
- ││0 @ 0 │0 @ 2 │1 @ 2 │ │
- ││ │axis 4 │axis 4 │ │
- │├──────┼───────────────┼──────────────────┤ │
- ││0 @ 1 │ 1.00 3.10e+1 │ 4.00 3.40e+1 │ │
- ││axis 3│ 2.00 3.20e+1 │ 5.00 3.50e+1 │ │
- ││ │ 3.00 3.30e+1 │ 6.00 3.60e+1 │ │
- │├──────┼───────────────┼──────────────────┤ │
- ││1 @ 1 │ 7.00 3.70e+1 │ 1.00e+1 4.00e+1 │ │
- ││axis 3│ 8.00 3.80e+1 │ 1.10e+1 4.10e+1 │ │
- ││ │ 9.00 3.90e+1 │ 1.20e+1 4.20e+1 │ │
- │└──────┴───────────────┴──────────────────┘ │
- ├──────────────────────────────────────────────────┤
- │┌──────┬──────────────────┬──────────────────┐ │
- ││1 @ 0 │0 @ 2 │1 @ 2 │ │
- ││ │axis 4 │axis 4 │ │
- │├──────┼──────────────────┼──────────────────┤ │
- ││0 @ 1 │ 1.30e+1 4.30e+1 │ 1.60e+1 4.60e+1 │ │
- ││axis 3│ 1.40e+1 4.40e+1 │ 1.70e+1 4.70e+1 │ │
- ││ │ 1.50e+1 4.50e+1 │ 1.80e+1 4.80e+1 │ │
- │├──────┼──────────────────┼──────────────────┤ │
- ││1 @ 1 │ 1.90e+1 4.90e+1 │ 2.20e+1 5.20e+1 │ │
- ││axis 3│ 2.00e+1 5.00e+1 │ 2.30e+1 5.30e+1 │ │
- ││ │ 2.10e+1 5.10e+1 │ 2.40e+1 5.40e+1 │ │
- │└──────┴──────────────────┴──────────────────┘ │
- └──────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let%op heyhoo4 =
[|
[
@@ -389,63 +194,22 @@ let%expect_test "Print constant tensor" =
in
Train.forward_and_forget backend ctx heyhoo4;
Tensor.print ~with_code:false ~with_grad:false `Inline @@ heyhoo4;
- [%expect
- {|
- [7]: c2x2x2x3x2_heyhoo4 shape 0:2|4:2->1:2,2:2,3:3 [|
- [
- [
- [ 1.00 , 31.00 ; 2.00 , 32.00 ; 3.00 , 33.00 ]
- ; [ 4.00 , 34.00 ; 5.00 , 35.00 ; 6.00 , 36.00 ]
- ]
- ; [
- [ 7.00 , 37.00 ; 8.00 , 38.00 ; 9.00 , 39.00 ]
- ; [ 10.00 , 40.00 ; 11.00 , 41.00 ; 12.00 , 42.00 ]
- ]
- ]
- ; [
- [
- [ 13.00 , 43.00 ; 14.00 , 44.00 ; 15.00 , 45.00 ]
- ; [ 16.00 , 46.00 ; 17.00 , 47.00 ; 18.00 , 48.00 ]
- ]
- ; [
- [ 19.00 , 49.00 ; 20.00 , 50.00 ; 21.00 , 51.00 ]
- ; [ 22.00 , 52.00 ; 23.00 , 53.00 ; 24.00 , 54.00 ]
- ]
- ]
- |]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ heyhoo4;
- [%expect
- {|
- ┌───────────────────────────────────────────────────┐
- │[7]: c2x2x2x3x2_heyhoo4 shape 0:2|4:2->1:2,2:2,3:3 │
- │┌──────┬───────────────┬──────────────────┐ │
- ││0 @ 0 │0 @ 2 │1 @ 2 │ │
- ││ │axis 4 │axis 4 │ │
- │├──────┼───────────────┼──────────────────┤ │
- ││0 @ 1 │ 1.00 3.10e+1 │ 4.00 3.40e+1 │ │
- ││axis 3│ 2.00 3.20e+1 │ 5.00 3.50e+1 │ │
- ││ │ 3.00 3.30e+1 │ 6.00 3.60e+1 │ │
- │├──────┼───────────────┼──────────────────┤ │
- ││1 @ 1 │ 7.00 3.70e+1 │ 1.00e+1 4.00e+1 │ │
- ││axis 3│ 8.00 3.80e+1 │ 1.10e+1 4.10e+1 │ │
- ││ │ 9.00 3.90e+1 │ 1.20e+1 4.20e+1 │ │
- │└──────┴───────────────┴──────────────────┘ │
- ├───────────────────────────────────────────────────┤
- │┌──────┬──────────────────┬──────────────────┐ │
- ││1 @ 0 │0 @ 2 │1 @ 2 │ │
- ││ │axis 4 │axis 4 │ │
- │├──────┼──────────────────┼──────────────────┤ │
- ││0 @ 1 │ 1.30e+1 4.30e+1 │ 1.60e+1 4.60e+1 │ │
- ││axis 3│ 1.40e+1 4.40e+1 │ 1.70e+1 4.70e+1 │ │
- ││ │ 1.50e+1 4.50e+1 │ 1.80e+1 4.80e+1 │ │
- │├──────┼──────────────────┼──────────────────┤ │
- ││1 @ 1 │ 1.90e+1 4.90e+1 │ 2.20e+1 5.20e+1 │ │
- ││axis 3│ 2.00e+1 5.00e+1 │ 2.30e+1 5.30e+1 │ │
- ││ │ 2.10e+1 5.10e+1 │ 2.40e+1 5.40e+1 │ │
- │└──────┴──────────────────┴──────────────────┘ │
- └───────────────────────────────────────────────────┘
- |}]
+ [%expect.unreachable]
+[@@expect.uncaught_exn {|
+ (* CR expect_test_collector: This test expectation appears to contain a backtrace.
+ This is strongly discouraged as backtraces are fragile.
+ Please change this test to not include a backtrace. *)
+ "Assert_failure arrayjit/lib/backends.ml:76:12"
+ Raised at Backends.Add_buffer_retrieval_and_syncing.update_writer_event.(fun) in file "arrayjit/lib/backends.ml", line 76, characters 12-35
+ Called from Base__Option.iter in file "src/option.ml" (inlined), line 83, characters 14-17
+ Called from Ir__Tnode.do_read in file "arrayjit/lib/tnode.ml", lines 597-601, characters 2-19
+ Called from Ocannl__Tensor.to_doc in file "lib/tensor.ml", line 666, characters 10-28
+ Called from Ocannl__Tensor.print in file "lib/tensor.ml", line 757, characters 4-63
+ Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 104, characters 2-63
+ Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28
+ |}]
let%expect_test "Matrix multiplication dims 2x3" =
Tensor.unsafe_reinitialize ();
@@ -464,31 +228,38 @@ let%expect_test "Matrix multiplication dims 2x3" =
let%op y = ("hey" 7.0 * [ 2; 3 ]) + [ 4; 5; 6 ] in
Train.forward_and_forget backend ctx y;
Tensor.print ~with_code:false ~with_grad:false `Default @@ hey;
- [%expect
- {|
- ┌────────────────────────┐
- │[0]: hey shape 1:2->0:3 │
- │┌──────┬────────────┐ │
- ││ │axis 1 │ │
- │├──────┼────────────┤│
- ││axis 0│ 7.00 7.00 │ │
- ││ │ 7.00 7.00 │ │
- ││ │ 7.00 7.00 │ │
- │└──────┴────────────┘ │
- └────────────────────────┘
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ y;
- [%expect
- {|
- ┌──────────────────────────────┐
- │[6]: +_y shape 0:3 │
- │┌┬───────────────────────────┐│
- │││axis 0 ││
- │├┼───────────────────────────┤│
- │││ 3.90e+1 4.00e+1 4.10e+1 ││
- │└┴───────────────────────────┘│
- └──────────────────────────────┘
- |}]
+ [%expect.unreachable]
+[@@expect.uncaught_exn {|
+ (* CR expect_test_collector: This test expectation appears to contain a backtrace.
+ This is strongly discouraged as backtraces are fragile.
+ Please change this test to not include a backtrace. *)
+ (Invalid_argument "index out of bounds")
+ Raised by primitive operation at Ir__Assignments.to_low_level.loop.(fun) in file "arrayjit/lib/assignments.ml", line 249, characters 39-54
+ Called from Ir__Low_level.unroll_dims.generate_all_combinations in file "arrayjit/lib/low_level.ml", lines 1191-1193, characters 12-40
+ Called from Ir__Low_level.unroll_dims.generate_all_combinations in file "arrayjit/lib/low_level.ml", lines 1191-1193, characters 12-40
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 227, characters 17-24
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 227, characters 17-24
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 228, characters 17-24
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 227, characters 17-24
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 225, characters 66-72
+ Called from Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", line 251, characters 2-11
+ Re-raised at Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", lines 130-251, characters 29-11
+ Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 414, characters 26-43
+ Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 414, characters 2-43
+ Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-75
+ Called from Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 193-194, characters 4-45
+ Re-raised at Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 185-194, characters 34-47
+ Called from Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", line 356, characters 6-61
+ Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 355-356, characters 4-61
+ Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 354-362, characters 26-99
+ Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 494, characters 37-90
+ Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 488-497, characters 32-17
+ Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 501, characters 12-75
+ Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 465, characters 2-40
+ Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28
+ |}]
let%expect_test "Big matrix" =
Tensor.unsafe_reinitialize ();
@@ -509,72 +280,41 @@ let%expect_test "Big matrix" =
let y = TDSL.O.((hey * zero_to_twenty) + zero_to_twenty) in
Train.forward_and_forget backend ctx y;
Tensor.print ~with_code:false ~with_grad:false `Inline zero_to_twenty;
- [%expect
- {|
- [2]: 0...20 shape 0:21 [
- 0.00
- ; 1.00
- ; 2.00
- ; 3.00
- ; 4.00
- ; 5.00
- ; 6.00
- ; 7.00
- ; 8.00
- ; 9.00
- ; 10.00
- ; 11.00
- ; 12.00
- ; 13.00
- ; 14.00
- ; 15.00
- ; 16.00
- ; 17.00
- ; 18.00
- ; 19.00
- ; 20.00
- ]
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default zero_to_twenty;
- [%expect
- {|
- ┌──────────────────────────────────────┐
- │[2]: 0...20 shape 0:21 │
- │┌┬───────────────────────────────────┐│
- │││axis 0 ││
- │├┼───────────────────────────────────┤│
- │││ 0.00 1.00 ... 1.90e+1 2.00e+1 ││
- │└┴───────────────────────────────────┘│
- └──────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default hey;
- [%expect
- {|
- ┌──────────────────────────────────────────────────┐
- │[0]: hey shape 1:21->0:21 │
- │┌──────┬─────────────────────────────────────────┐│
- ││ │axis 1 ││
- │├──────┼─────────────────────────────────────────┤│
- ││axis 0│ 5.00e-1 5.00e-1 ... 5.00e-1 5.00e-1 ││
- ││ │ 5.00e-1 5.00e-1 ... 5.00e-1 5.00e-1 ││
- ││ │ ... ... ... ... ... ││
- ││ │ 5.00e-1 5.00e-1 ... 5.00e-1 5.00e-1 ││
- ││ │ 5.00e-1 5.00e-1 ... 5.00e-1 5.00e-1 ││
- │└──────┴─────────────────────────────────────────┘│
- └──────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default y;
- [%expect
- {|
- ┌────────────────────────────────────────────┐
- │[5]: + shape 0:21 │
- │┌┬─────────────────────────────────────────┐│
- │││axis 0 ││
- │├┼─────────────────────────────────────────┤│
- │││ 1.05e+2 1.06e+2 ... 1.24e+2 1.25e+2 ││
- │└┴─────────────────────────────────────────┘│
- └────────────────────────────────────────────┘
- |}]
+ [%expect.unreachable]
+[@@expect.uncaught_exn {|
+ (* CR expect_test_collector: This test expectation appears to contain a backtrace.
+ This is strongly discouraged as backtraces are fragile.
+ Please change this test to not include a backtrace. *)
+ (Invalid_argument "index out of bounds")
+ Raised by primitive operation at Ir__Assignments.to_low_level.loop.(fun) in file "arrayjit/lib/assignments.ml", line 249, characters 39-54
+ Called from Ir__Low_level.unroll_dims.generate_all_combinations in file "arrayjit/lib/low_level.ml", lines 1191-1193, characters 12-40
+ Called from Ir__Low_level.unroll_dims.generate_all_combinations in file "arrayjit/lib/low_level.ml", lines 1191-1193, characters 12-40
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 227, characters 17-24
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 227, characters 17-24
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 227, characters 17-24
+ Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 225, characters 66-72
+ Called from Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", line 251, characters 2-11
+ Re-raised at Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", lines 130-251, characters 29-11
+ Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 414, characters 26-43
+ Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 414, characters 2-43
+ Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-75
+ Called from Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 193-194, characters 4-45
+ Re-raised at Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 185-194, characters 34-47
+ Called from Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", line 356, characters 6-61
+ Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 355-356, characters 4-61
+ Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 354-362, characters 26-99
+ Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 494, characters 37-90
+ Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 488-497, characters 32-17
+ Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 501, characters 12-75
+ Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 510, characters 2-40
+ Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28
+ |}]
let%expect_test "Very big tensor" =
Tensor.unsafe_reinitialize ();
@@ -596,140 +336,7 @@ let%expect_test "Very big tensor" =
Train.forward_and_forget backend ctx hoo;
Tensor.print ~with_code:false ~with_grad:false `Default hey;
[%expect
- {|
- ┌───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[0]: r6x10x11x7x8x9 shape 0:6|3:7,4:8,5:9->1:10,2:11 │
- │┌──────┬─────────────────────────────────────────┬─────────────────────────────────────────┬──────┬─────────────────────────────────────────┬─────────────────────────────────────────┐│
- ││0 @ 0 │0 @ 4 │1 @ 4 │~~~~~ │6 @ 4 │7 @ 4 ││
- ││ │axis 5 │axis 5 │axis 5│axis 5 │axis 5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││0 @ 1 │ 0.00 1.00 ... 7.00 8.00 │ 9.00 1.00e+1 ... 1.60e+1 1.70e+1 │ ... │ 5.40e+1 5.50e+1 ... 6.10e+1 6.20e+1 │ 6.30e+1 6.40e+1 ... 7.00e+1 7.10e+1 ││
- ││axis 2│ 5.04e+2 5.05e+2 ... 5.11e+2 5.12e+2 │ 5.13e+2 5.14e+2 ... 5.20e+2 5.21e+2 │ │ 5.58e+2 5.59e+2 ... 5.65e+2 5.66e+2 │ 5.67e+2 5.68e+2 ... 5.74e+2 5.75e+2 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 4.53e+3 4.53e+3 ... 4.54e+3 4.54e+3 │ 4.54e+3 4.54e+3 ... 4.55e+3 4.55e+3 │ │ 4.59e+3 4.59e+3 ... 4.59e+3 4.59e+3 │ 4.59e+3 4.60e+3 ... 4.60e+3 4.60e+3 ││
- ││ │ 5.04e+3 5.04e+3 ... 5.04e+3 5.04e+3 │ 5.04e+3 5.05e+3 ... 5.05e+3 5.05e+3 │ │ 5.09e+3 5.09e+3 ... 5.10e+3 5.10e+3 │ 5.10e+3 5.10e+3 ... 5.11e+3 5.11e+3 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││1 @ 1 │ 5.54e+3 5.54e+3 ... 5.55e+3 5.55e+3 │ 5.55e+3 5.55e+3 ... 5.56e+3 5.56e+3 │ ... │ 5.59e+3 5.59e+3 ... 5.60e+3 5.60e+3 │ 5.60e+3 5.60e+3 ... 5.61e+3 5.61e+3 ││
- ││axis 2│ 6.04e+3 6.04e+3 ... 6.05e+3 6.05e+3 │ 6.05e+3 6.05e+3 ... 6.06e+3 6.06e+3 │ │ 6.10e+3 6.10e+3 ... 6.10e+3 6.11e+3 │ 6.11e+3 6.11e+3 ... 6.11e+3 6.11e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.00e+4 1.00e+4 ... 1.00e+4 1.00e+4 │ 1.00e+4 1.00e+4 ... 1.00e+4 1.00e+4 │ │ 1.01e+4 1.01e+4 ... 1.01e+4 1.01e+4 │ 1.01e+4 1.01e+4 ... 1.01e+4 1.01e+4 ││
- ││ │ 1.05e+4 1.05e+4 ... 1.05e+4 1.05e+4 │ 1.05e+4 1.05e+4 ... 1.06e+4 1.06e+4 │ │ 1.06e+4 1.06e+4 ... 1.06e+4 1.06e+4 │ 1.06e+4 1.06e+4 ... 1.06e+4 1.06e+4 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││~~~~~ │ ... │ ... │ ... │ ... │ ... ││
- ││axis 2│ │ │ │ │ ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││8 @ 1 │ 4.43e+4 4.43e+4 ... 4.43e+4 4.43e+4 │ 4.43e+4 4.43e+4 ... 4.43e+4 4.43e+4 │ ... │ 4.44e+4 4.44e+4 ... 4.44e+4 4.44e+4 │ 4.44e+4 4.44e+4 ... 4.44e+4 4.44e+4 ││
- ││axis 2│ 4.48e+4 4.48e+4 ... 4.48e+4 4.48e+4 │ 4.48e+4 4.48e+4 ... 4.48e+4 4.48e+4 │ │ 4.49e+4 4.49e+4 ... 4.49e+4 4.49e+4 │ 4.49e+4 4.49e+4 ... 4.49e+4 4.49e+4 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 4.88e+4 4.88e+4 ... 4.88e+4 4.88e+4 │ 4.88e+4 4.88e+4 ... 4.89e+4 4.89e+4 │ │ 4.89e+4 4.89e+4 ... 4.89e+4 4.89e+4 │ 4.89e+4 4.89e+4 ... 4.89e+4 4.89e+4 ││
- ││ │ 4.93e+4 4.93e+4 ... 4.93e+4 4.94e+4 │ 4.94e+4 4.94e+4 ... 4.94e+4 4.94e+4 │ │ 4.94e+4 4.94e+4 ... 4.94e+4 4.94e+4 │ 4.94e+4 4.94e+4 ... 4.94e+4 4.94e+4 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││9 @ 1 │ 4.98e+4 4.98e+4 ... 4.99e+4 4.99e+4 │ 4.99e+4 4.99e+4 ... 4.99e+4 4.99e+4 │ ... │ 4.99e+4 4.99e+4 ... 4.99e+4 4.99e+4 │ 4.99e+4 4.99e+4 ... 4.99e+4 4.99e+4 ││
- ││axis 2│ 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 │ 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 │ │ 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 │ 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 5.44e+4 5.44e+4 ... 5.44e+4 5.44e+4 │ 5.44e+4 5.44e+4 ... 5.44e+4 5.44e+4 │ │ 5.44e+4 5.44e+4 ... 5.44e+4 5.44e+4 │ 5.44e+4 5.44e+4 ... 5.45e+4 5.45e+4 ││
- ││ │ 5.49e+4 5.49e+4 ... 5.49e+4 5.49e+4 │ 5.49e+4 5.49e+4 ... 5.49e+4 5.49e+4 │ │ 5.49e+4 5.49e+4 ... 5.49e+4 5.49e+4 │ 5.49e+4 5.50e+4 ... 5.50e+4 5.50e+4 ││
- │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴──────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│
- ├───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
- │┌──────┬─────────────────────────────────────────┬─────────────────────────────────────────┬──────┬─────────────────────────────────────────┬─────────────────────────────────────────┐│
- ││1 @ 0 │0 @ 4 │1 @ 4 │~~~~~ │6 @ 4 │7 @ 4 ││
- ││ │axis 5 │axis 5 │axis 5│axis 5 │axis 5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││0 @ 1 │ 5.54e+4 5.54e+4 ... 5.54e+4 5.54e+4 │ 5.54e+4 5.54e+4 ... 5.54e+4 5.54e+4 │ ... │ 5.54e+4 5.54e+4 ... 5.55e+4 5.55e+4 │ 5.55e+4 5.55e+4 ... 5.55e+4 5.55e+4 ││
- ││axis 2│ 5.59e+4 5.59e+4 ... 5.59e+4 5.59e+4 │ 5.59e+4 5.59e+4 ... 5.59e+4 5.59e+4 │ │ 5.59e+4 5.59e+4 ... 5.60e+4 5.60e+4 │ 5.60e+4 5.60e+4 ... 5.60e+4 5.60e+4 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 5.99e+4 5.99e+4 ... 5.99e+4 5.99e+4 │ 5.99e+4 5.99e+4 ... 5.99e+4 5.99e+4 │ │ 6.00e+4 6.00e+4 ... 6.00e+4 6.00e+4 │ 6.00e+4 6.00e+4 ... 6.00e+4 6.00e+4 ││
- ││ │ 6.04e+4 6.04e+4 ... 6.04e+4 6.04e+4 │ 6.04e+4 6.04e+4 ... 6.04e+4 6.04e+4 │ │ 6.05e+4 6.05e+4 ... 6.05e+4 6.05e+4 │ 6.05e+4 6.05e+4 ... 6.05e+4 6.05e+4 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││1 @ 1 │ 6.09e+4 6.09e+4 ... 6.09e+4 6.09e+4 │ 6.09e+4 6.09e+4 ... 6.10e+4 6.10e+4 │ ... │ 6.10e+4 6.10e+4 ... 6.10e+4 6.10e+4 │ 6.10e+4 6.10e+4 ... 6.10e+4 6.10e+4 ││
- ││axis 2│ 6.14e+4 6.14e+4 ... 6.14e+4 6.14e+4 │ 6.14e+4 6.14e+4 ... 6.15e+4 6.15e+4 │ │ 6.15e+4 6.15e+4 ... 6.15e+4 6.15e+4 │ 6.15e+4 6.15e+4 ... 6.15e+4 6.15e+4 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 │ 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 │ │ 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 │ 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 ││
- ││ │ 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 │ 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 │ │ 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 │ 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││~~~~~ │ ... │ ... │ ... │ ... │ ... ││
- ││axis 2│ │ │ │ │ ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││8 @ 1 │ 9.97e+4 9.97e+4 ... 9.97e+4 9.98e+4 │ 9.98e+4 9.98e+4 ... 9.98e+4 9.98e+4 │ ... │ 9.98e+4 9.98e+4 ... 9.98e+4 9.98e+4 │ 9.98e+4 9.98e+4 ... 9.98e+4 9.98e+4 ││
- ││axis 2│ 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 │ 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 │ │ 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 │ 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 │ │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 ││
- ││ │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 │ │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 │ 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││9 @ 1 │ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 │ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 │ ... │ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 │ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 ││
- ││axis 2│ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 │ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 │ │ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 │ 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 │ 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 │ │ 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 │ 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 ││
- ││ │ 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 │ 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 │ │ 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 │ 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 ││
- │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴──────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│
- ├───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
- │ ... │
- ├───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
- │┌──────┬─────────────────────────────────────────┬─────────────────────────────────────────┬──────┬─────────────────────────────────────────┬─────────────────────────────────────────┐│
- ││4 @ 0 │0 @ 4 │1 @ 4 │~~~~~ │6 @ 4 │7 @ 4 ││
- ││ │axis 5 │axis 5 │axis 5│axis 5 │axis 5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││0 @ 1 │ 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 │ 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 │ ... │ 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 │ 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 ││
- ││axis 2│ 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 │ 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 │ │ 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 │ 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 │ │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 ││
- ││ │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 │ │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 │ 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││1 @ 1 │ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 │ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 │ ... │ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 │ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 ││
- ││axis 2│ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 │ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 │ │ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 │ 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 │ 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 │ │ 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 │ 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 ││
- ││ │ 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 │ 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 │ │ 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 │ 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││~~~~~ │ ... │ ... │ ... │ ... │ ... ││
- ││axis 2│ │ │ │ │ ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││8 @ 1 │ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 │ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 │ ... │ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 │ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 ││
- ││axis 2│ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 │ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 │ │ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 │ 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 │ 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 │ │ 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 │ 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 ││
- ││ │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 │ │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││9 @ 1 │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 │ ... │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 │ 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 ││
- ││axis 2│ 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 │ 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 │ │ 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 │ 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 │ │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 ││
- ││ │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 │ │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 │ 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 ││
- │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴──────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│
- ├───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
- │┌──────┬─────────────────────────────────────────┬─────────────────────────────────────────┬──────┬─────────────────────────────────────────┬─────────────────────────────────────────┐│
- ││5 @ 0 │0 @ 4 │1 @ 4 │~~~~~ │6 @ 4 │7 @ 4 ││
- ││ │axis 5 │axis 5 │axis 5│axis 5 │axis 5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││0 @ 1 │ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 │ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 │ ... │ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 │ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 ││
- ││axis 2│ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 │ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 │ │ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 │ 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 │ 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 │ │ 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 │ 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 ││
- ││ │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 │ │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││1 @ 1 │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 │ ... │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 │ 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 ││
- ││axis 2│ 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 │ 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 │ │ 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 │ 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 │ │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 ││
- ││ │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 │ │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 │ 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││~~~~~ │ ... │ ... │ ... │ ... │ ... ││
- ││axis 2│ │ │ │ │ ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││8 @ 1 │ 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 │ 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 │ ... │ 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 │ 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 ││
- ││axis 2│ 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 │ 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 │ │ 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 │ 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 │ │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 ││
- ││ │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 │ │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 │ 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││9 @ 1 │ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 │ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 │ ... │ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 │ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 ││
- ││axis 2│ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 │ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 │ │ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 │ 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 │ 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 │ │ 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 │ 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 ││
- ││ │ 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 │ 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 │ │ 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 │ 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 ││
- │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴──────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│
- └───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ {| [0]: r6x10x11x7x8x9 shape 0:6|3:7,4:8,5:9->1:10,2:11 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default hoo;
(* Disable line wrapping for viewing the output. In VSCode: `View: Toggle Word Wrap`. *)
[%expect
File "test/einsum_trivia.ml", line 1, characters 0-0:
/usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum_trivia.ml _build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/einsum_trivia.ml.corrected
diff --git a/_build/default/test/einsum_trivia.ml b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/einsum_trivia.ml.corrected
index 162aa57..7549686 100644
--- a/_build/default/test/einsum_trivia.ml
+++ b/_build/.sandbox/206e79d9026ef941d97a5bbc9a8bd5b0/default/test/einsum_trivia.ml.corrected
@@ -27,20 +27,7 @@ let%expect_test "einsum1 permute axes" =
Train.forward_and_forget backend ctx ho;
Tensor.print ~with_code:false ~with_grad:false `Default @@ hey;
[%expect
- {|
- ┌─────────────────────────────────────────────────────────────┐
- │[0]: r2x4x3 shape 0:2|2:3->1:4 │
- │┌──────┬────────────────────────┬───────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 ││
- ││ │axis 2 │axis 2 ││
- │├──────┼────────────────────────┼───────────────────────────┤│
- ││axis 1│ 0.00 1.00 2.00 │ 1.20e+1 1.30e+1 1.40e+1 ││
- ││ │ 3.00 4.00 5.00 │ 1.50e+1 1.60e+1 1.70e+1 ││
- ││ │ 6.00 7.00 8.00 │ 1.80e+1 1.90e+1 2.00e+1 ││
- ││ │ 9.00 1.00e+1 1.10e+1 │ 2.10e+1 2.20e+1 2.30e+1 ││
- │└──────┴────────────────────────┴───────────────────────────┘│
- └─────────────────────────────────────────────────────────────┘
- |}];
+ {| [0]: r2x4x3 shape 0:2|2:3->1:4 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ ho;
[%expect
{|
@@ -63,106 +50,7 @@ let%expect_test "einsum1 permute axes" =
Train.forward_and_forget backend ctx ho2;
Tensor.print ~with_code:false ~with_grad:false `Default @@ hey2;
[%expect
- {|
- ┌────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[2]: r2x3x6x7x4x5 shape 0:2,1:3|4:4,5:5->2:6,3:7 │
- │┌──────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┐│
- ││0 @ 1 │0 @ 4 │1 @ 4 │2 @ 4 │3 @ 4 ││
- ││ │axis 5 │axis 5 │axis 5 │axis 5 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││0 @ 2 │ 0.00 1.00 2.00 3.00 4.00 │ 5.00 6.00 7.00 8.00 9.00 │ 1.00e+1 1.10e+1 1.20e+1 1.30e+1 1.40e+1 │ 1.50e+1 1.60e+1 1.70e+1 1.80e+1 1.90e+1 ││
- ││axis 3│ 2.00e+1 2.10e+1 2.20e+1 2.30e+1 2.40e+1 │ 2.50e+1 2.60e+1 2.70e+1 2.80e+1 2.90e+1 │ 3.00e+1 3.10e+1 3.20e+1 3.30e+1 3.40e+1 │ 3.50e+1 3.60e+1 3.70e+1 3.80e+1 3.90e+1 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.00e+2 1.01e+2 1.02e+2 1.03e+2 1.04e+2 │ 1.05e+2 1.06e+2 1.07e+2 1.08e+2 1.09e+2 │ 1.10e+2 1.11e+2 1.12e+2 1.13e+2 1.14e+2 │ 1.15e+2 1.16e+2 1.17e+2 1.18e+2 1.19e+2 ││
- ││ │ 1.20e+2 1.21e+2 1.22e+2 1.23e+2 1.24e+2 │ 1.25e+2 1.26e+2 1.27e+2 1.28e+2 1.29e+2 │ 1.30e+2 1.31e+2 1.32e+2 1.33e+2 1.34e+2 │ 1.35e+2 1.36e+2 1.37e+2 1.38e+2 1.39e+2 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││1 @ 2 │ 1.40e+2 1.41e+2 1.42e+2 1.43e+2 1.44e+2 │ 1.45e+2 1.46e+2 1.47e+2 1.48e+2 1.49e+2 │ 1.50e+2 1.51e+2 1.52e+2 1.53e+2 1.54e+2 │ 1.55e+2 1.56e+2 1.57e+2 1.58e+2 1.59e+2 ││
- ││axis 3│ 1.60e+2 1.61e+2 1.62e+2 1.63e+2 1.64e+2 │ 1.65e+2 1.66e+2 1.67e+2 1.68e+2 1.69e+2 │ 1.70e+2 1.71e+2 1.72e+2 1.73e+2 1.74e+2 │ 1.75e+2 1.76e+2 1.77e+2 1.78e+2 1.79e+2 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.40e+2 2.41e+2 2.42e+2 2.43e+2 2.44e+2 │ 2.45e+2 2.46e+2 2.47e+2 2.48e+2 2.49e+2 │ 2.50e+2 2.51e+2 2.52e+2 2.53e+2 2.54e+2 │ 2.55e+2 2.56e+2 2.57e+2 2.58e+2 2.59e+2 ││
- ││ │ 2.60e+2 2.61e+2 2.62e+2 2.63e+2 2.64e+2 │ 2.65e+2 2.66e+2 2.67e+2 2.68e+2 2.69e+2 │ 2.70e+2 2.71e+2 2.72e+2 2.73e+2 2.74e+2 │ 2.75e+2 2.76e+2 2.77e+2 2.78e+2 2.79e+2 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││~~~~~ │ ... │ ... │ ... │ ... ││
- ││axis 3│ │ │ │ ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││4 @ 2 │ 5.60e+2 5.61e+2 5.62e+2 5.63e+2 5.64e+2 │ 5.65e+2 5.66e+2 5.67e+2 5.68e+2 5.69e+2 │ 5.70e+2 5.71e+2 5.72e+2 5.73e+2 5.74e+2 │ 5.75e+2 5.76e+2 5.77e+2 5.78e+2 5.79e+2 ││
- ││axis 3│ 5.80e+2 5.81e+2 5.82e+2 5.83e+2 5.84e+2 │ 5.85e+2 5.86e+2 5.87e+2 5.88e+2 5.89e+2 │ 5.90e+2 5.91e+2 5.92e+2 5.93e+2 5.94e+2 │ 5.95e+2 5.96e+2 5.97e+2 5.98e+2 5.99e+2 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 6.60e+2 6.61e+2 6.62e+2 6.63e+2 6.64e+2 │ 6.65e+2 6.66e+2 6.67e+2 6.68e+2 6.69e+2 │ 6.70e+2 6.71e+2 6.72e+2 6.73e+2 6.74e+2 │ 6.75e+2 6.76e+2 6.77e+2 6.78e+2 6.79e+2 ││
- ││ │ 6.80e+2 6.81e+2 6.82e+2 6.83e+2 6.84e+2 │ 6.85e+2 6.86e+2 6.87e+2 6.88e+2 6.89e+2 │ 6.90e+2 6.91e+2 6.92e+2 6.93e+2 6.94e+2 │ 6.95e+2 6.96e+2 6.97e+2 6.98e+2 6.99e+2 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││5 @ 2 │ 7.00e+2 7.01e+2 7.02e+2 7.03e+2 7.04e+2 │ 7.05e+2 7.06e+2 7.07e+2 7.08e+2 7.09e+2 │ 7.10e+2 7.11e+2 7.12e+2 7.13e+2 7.14e+2 │ 7.15e+2 7.16e+2 7.17e+2 7.18e+2 7.19e+2 ││
- ││axis 3│ 7.20e+2 7.21e+2 7.22e+2 7.23e+2 7.24e+2 │ 7.25e+2 7.26e+2 7.27e+2 7.28e+2 7.29e+2 │ 7.30e+2 7.31e+2 7.32e+2 7.33e+2 7.34e+2 │ 7.35e+2 7.36e+2 7.37e+2 7.38e+2 7.39e+2 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 8.00e+2 8.01e+2 8.02e+2 8.03e+2 8.04e+2 │ 8.05e+2 8.06e+2 8.07e+2 8.08e+2 8.09e+2 │ 8.10e+2 8.11e+2 8.12e+2 8.13e+2 8.14e+2 │ 8.15e+2 8.16e+2 8.17e+2 8.18e+2 8.19e+2 ││
- ││ │ 8.20e+2 8.21e+2 8.22e+2 8.23e+2 8.24e+2 │ 8.25e+2 8.26e+2 8.27e+2 8.28e+2 8.29e+2 │ 8.30e+2 8.31e+2 8.32e+2 8.33e+2 8.34e+2 │ 8.35e+2 8.36e+2 8.37e+2 8.38e+2 8.39e+2 ││
- │└──────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┘│
- ├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
- │┌──────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┐│
- ││1 @ 1 │0 @ 4 │1 @ 4 │2 @ 4 │3 @ 4 ││
- ││ │axis 5 │axis 5 │axis 5 │axis 5 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││0 @ 2 │ 8.40e+2 8.41e+2 8.42e+2 8.43e+2 8.44e+2 │ 8.45e+2 8.46e+2 8.47e+2 8.48e+2 8.49e+2 │ 8.50e+2 8.51e+2 8.52e+2 8.53e+2 8.54e+2 │ 8.55e+2 8.56e+2 8.57e+2 8.58e+2 8.59e+2 ││
- ││axis 3│ 8.60e+2 8.61e+2 8.62e+2 8.63e+2 8.64e+2 │ 8.65e+2 8.66e+2 8.67e+2 8.68e+2 8.69e+2 │ 8.70e+2 8.71e+2 8.72e+2 8.73e+2 8.74e+2 │ 8.75e+2 8.76e+2 8.77e+2 8.78e+2 8.79e+2 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 9.40e+2 9.41e+2 9.42e+2 9.43e+2 9.44e+2 │ 9.45e+2 9.46e+2 9.47e+2 9.48e+2 9.49e+2 │ 9.50e+2 9.51e+2 9.52e+2 9.53e+2 9.54e+2 │ 9.55e+2 9.56e+2 9.57e+2 9.58e+2 9.59e+2 ││
- ││ │ 9.60e+2 9.61e+2 9.62e+2 9.63e+2 9.64e+2 │ 9.65e+2 9.66e+2 9.67e+2 9.68e+2 9.69e+2 │ 9.70e+2 9.71e+2 9.72e+2 9.73e+2 9.74e+2 │ 9.75e+2 9.76e+2 9.77e+2 9.78e+2 9.79e+2 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││1 @ 2 │ 9.80e+2 9.81e+2 9.82e+2 9.83e+2 9.84e+2 │ 9.85e+2 9.86e+2 9.87e+2 9.88e+2 9.89e+2 │ 9.90e+2 9.91e+2 9.92e+2 9.93e+2 9.94e+2 │ 9.95e+2 9.96e+2 9.97e+2 9.98e+2 9.99e+2 ││
- ││axis 3│ 1.00e+3 1.00e+3 1.00e+3 1.00e+3 1.00e+3 │ 1.00e+3 1.00e+3 1.00e+3 1.00e+3 1.00e+3 │ 1.01e+3 1.01e+3 1.01e+3 1.01e+3 1.01e+3 │ 1.01e+3 1.01e+3 1.01e+3 1.01e+3 1.01e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.08e+3 1.08e+3 1.08e+3 1.08e+3 1.08e+3 │ 1.08e+3 1.08e+3 1.08e+3 1.08e+3 1.08e+3 │ 1.09e+3 1.09e+3 1.09e+3 1.09e+3 1.09e+3 │ 1.09e+3 1.09e+3 1.09e+3 1.09e+3 1.09e+3 ││
- ││ │ 1.10e+3 1.10e+3 1.10e+3 1.10e+3 1.10e+3 │ 1.10e+3 1.10e+3 1.10e+3 1.10e+3 1.10e+3 │ 1.11e+3 1.11e+3 1.11e+3 1.11e+3 1.11e+3 │ 1.11e+3 1.11e+3 1.11e+3 1.11e+3 1.11e+3 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││~~~~~ │ ... │ ... │ ... │ ... ││
- ││axis 3│ │ │ │ ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││4 @ 2 │ 1.40e+3 1.40e+3 1.40e+3 1.40e+3 1.40e+3 │ 1.40e+3 1.40e+3 1.40e+3 1.40e+3 1.40e+3 │ 1.41e+3 1.41e+3 1.41e+3 1.41e+3 1.41e+3 │ 1.41e+3 1.41e+3 1.41e+3 1.41e+3 1.41e+3 ││
- ││axis 3│ 1.42e+3 1.42e+3 1.42e+3 1.42e+3 1.42e+3 │ 1.42e+3 1.42e+3 1.42e+3 1.42e+3 1.42e+3 │ 1.43e+3 1.43e+3 1.43e+3 1.43e+3 1.43e+3 │ 1.43e+3 1.43e+3 1.43e+3 1.43e+3 1.43e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.50e+3 1.50e+3 1.50e+3 1.50e+3 1.50e+3 │ 1.50e+3 1.50e+3 1.50e+3 1.50e+3 1.50e+3 │ 1.51e+3 1.51e+3 1.51e+3 1.51e+3 1.51e+3 │ 1.51e+3 1.51e+3 1.51e+3 1.51e+3 1.51e+3 ││
- ││ │ 1.52e+3 1.52e+3 1.52e+3 1.52e+3 1.52e+3 │ 1.52e+3 1.52e+3 1.52e+3 1.52e+3 1.52e+3 │ 1.53e+3 1.53e+3 1.53e+3 1.53e+3 1.53e+3 │ 1.53e+3 1.53e+3 1.53e+3 1.53e+3 1.53e+3 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││5 @ 2 │ 1.54e+3 1.54e+3 1.54e+3 1.54e+3 1.54e+3 │ 1.54e+3 1.54e+3 1.54e+3 1.54e+3 1.54e+3 │ 1.55e+3 1.55e+3 1.55e+3 1.55e+3 1.55e+3 │ 1.55e+3 1.55e+3 1.55e+3 1.55e+3 1.55e+3 ││
- ││axis 3│ 1.56e+3 1.56e+3 1.56e+3 1.56e+3 1.56e+3 │ 1.56e+3 1.56e+3 1.56e+3 1.56e+3 1.56e+3 │ 1.57e+3 1.57e+3 1.57e+3 1.57e+3 1.57e+3 │ 1.57e+3 1.57e+3 1.57e+3 1.57e+3 1.57e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.64e+3 1.64e+3 1.64e+3 1.64e+3 1.64e+3 │ 1.64e+3 1.64e+3 1.64e+3 1.64e+3 1.64e+3 │ 1.65e+3 1.65e+3 1.65e+3 1.65e+3 1.65e+3 │ 1.65e+3 1.65e+3 1.65e+3 1.65e+3 1.65e+3 ││
- ││ │ 1.66e+3 1.66e+3 1.66e+3 1.66e+3 1.66e+3 │ 1.66e+3 1.66e+3 1.66e+3 1.66e+3 1.66e+3 │ 1.67e+3 1.67e+3 1.67e+3 1.67e+3 1.67e+3 │ 1.67e+3 1.67e+3 1.67e+3 1.67e+3 1.67e+3 ││
- │└──────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┘│
- ├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
- │┌──────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┬─────────────────────────────────────────────┐│
- ││2 @ 1 │0 @ 4 │1 @ 4 │2 @ 4 │3 @ 4 ││
- ││ │axis 5 │axis 5 │axis 5 │axis 5 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││0 @ 2 │ 1.68e+3 1.68e+3 1.68e+3 1.68e+3 1.68e+3 │ 1.68e+3 1.68e+3 1.68e+3 1.68e+3 1.68e+3 │ 1.69e+3 1.69e+3 1.69e+3 1.69e+3 1.69e+3 │ 1.69e+3 1.69e+3 1.69e+3 1.69e+3 1.69e+3 ││
- ││axis 3│ 1.70e+3 1.70e+3 1.70e+3 1.70e+3 1.70e+3 │ 1.70e+3 1.70e+3 1.70e+3 1.70e+3 1.70e+3 │ 1.71e+3 1.71e+3 1.71e+3 1.71e+3 1.71e+3 │ 1.71e+3 1.71e+3 1.71e+3 1.71e+3 1.71e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.78e+3 1.78e+3 1.78e+3 1.78e+3 1.78e+3 │ 1.78e+3 1.78e+3 1.78e+3 1.78e+3 1.78e+3 │ 1.79e+3 1.79e+3 1.79e+3 1.79e+3 1.79e+3 │ 1.79e+3 1.79e+3 1.79e+3 1.79e+3 1.79e+3 ││
- ││ │ 1.80e+3 1.80e+3 1.80e+3 1.80e+3 1.80e+3 │ 1.80e+3 1.80e+3 1.80e+3 1.80e+3 1.80e+3 │ 1.81e+3 1.81e+3 1.81e+3 1.81e+3 1.81e+3 │ 1.81e+3 1.81e+3 1.81e+3 1.81e+3 1.81e+3 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││1 @ 2 │ 1.82e+3 1.82e+3 1.82e+3 1.82e+3 1.82e+3 │ 1.82e+3 1.82e+3 1.82e+3 1.82e+3 1.82e+3 │ 1.83e+3 1.83e+3 1.83e+3 1.83e+3 1.83e+3 │ 1.83e+3 1.83e+3 1.83e+3 1.83e+3 1.83e+3 ││
- ││axis 3│ 1.84e+3 1.84e+3 1.84e+3 1.84e+3 1.84e+3 │ 1.84e+3 1.84e+3 1.84e+3 1.84e+3 1.84e+3 │ 1.85e+3 1.85e+3 1.85e+3 1.85e+3 1.85e+3 │ 1.85e+3 1.85e+3 1.85e+3 1.85e+3 1.85e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 1.92e+3 1.92e+3 1.92e+3 1.92e+3 1.92e+3 │ 1.92e+3 1.92e+3 1.92e+3 1.92e+3 1.92e+3 │ 1.93e+3 1.93e+3 1.93e+3 1.93e+3 1.93e+3 │ 1.93e+3 1.93e+3 1.93e+3 1.93e+3 1.93e+3 ││
- ││ │ 1.94e+3 1.94e+3 1.94e+3 1.94e+3 1.94e+3 │ 1.94e+3 1.94e+3 1.94e+3 1.94e+3 1.94e+3 │ 1.95e+3 1.95e+3 1.95e+3 1.95e+3 1.95e+3 │ 1.95e+3 1.95e+3 1.95e+3 1.95e+3 1.95e+3 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││~~~~~ │ ... │ ... │ ... │ ... ││
- ││axis 3│ │ │ │ ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││4 @ 2 │ 2.24e+3 2.24e+3 2.24e+3 2.24e+3 2.24e+3 │ 2.24e+3 2.24e+3 2.24e+3 2.24e+3 2.24e+3 │ 2.25e+3 2.25e+3 2.25e+3 2.25e+3 2.25e+3 │ 2.25e+3 2.25e+3 2.25e+3 2.25e+3 2.25e+3 ││
- ││axis 3│ 2.26e+3 2.26e+3 2.26e+3 2.26e+3 2.26e+3 │ 2.26e+3 2.26e+3 2.26e+3 2.26e+3 2.26e+3 │ 2.27e+3 2.27e+3 2.27e+3 2.27e+3 2.27e+3 │ 2.27e+3 2.27e+3 2.27e+3 2.27e+3 2.27e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.34e+3 2.34e+3 2.34e+3 2.34e+3 2.34e+3 │ 2.34e+3 2.34e+3 2.34e+3 2.34e+3 2.34e+3 │ 2.35e+3 2.35e+3 2.35e+3 2.35e+3 2.35e+3 │ 2.35e+3 2.35e+3 2.35e+3 2.35e+3 2.35e+3 ││
- ││ │ 2.36e+3 2.36e+3 2.36e+3 2.36e+3 2.36e+3 │ 2.36e+3 2.36e+3 2.36e+3 2.36e+3 2.36e+3 │ 2.37e+3 2.37e+3 2.37e+3 2.37e+3 2.37e+3 │ 2.37e+3 2.37e+3 2.37e+3 2.37e+3 2.37e+3 ││
- │├──────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┼─────────────────────────────────────────────┤│
- ││5 @ 2 │ 2.38e+3 2.38e+3 2.38e+3 2.38e+3 2.38e+3 │ 2.38e+3 2.38e+3 2.38e+3 2.38e+3 2.38e+3 │ 2.39e+3 2.39e+3 2.39e+3 2.39e+3 2.39e+3 │ 2.39e+3 2.39e+3 2.39e+3 2.39e+3 2.39e+3 ││
- ││axis 3│ 2.40e+3 2.40e+3 2.40e+3 2.40e+3 2.40e+3 │ 2.40e+3 2.40e+3 2.40e+3 2.40e+3 2.40e+3 │ 2.41e+3 2.41e+3 2.41e+3 2.41e+3 2.41e+3 │ 2.41e+3 2.41e+3 2.41e+3 2.41e+3 2.41e+3 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 2.48e+3 2.48e+3 2.48e+3 2.48e+3 2.48e+3 │ 2.48e+3 2.48e+3 2.48e+3 2.48e+3 2.48e+3 │ 2.49e+3 2.49e+3 2.49e+3 2.49e+3 2.49e+3 │ 2.49e+3 2.49e+3 2.49e+3 2.49e+3 2.49e+3 ││
- ││ │ 2.50e+3 2.50e+3 2.50e+3 2.50e+3 2.50e+3 │ 2.50e+3 2.50e+3 2.50e+3 2.50e+3 2.50e+3 │ 2.51e+3 2.51e+3 2.51e+3 2.51e+3 2.51e+3 │ 2.51e+3 2.51e+3 2.51e+3 2.51e+3 2.51e+3 ││
- │└──────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┴─────────────────────────────────────────────┘│
- └────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ {| [2]: r2x3x6x7x4x5 shape 0:2,1:3|4:4,5:5->2:6,3:7 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ ho2;
[%expect
{|
@@ -290,20 +178,7 @@ let%expect_test "einsum1 sum out axes" =
Train.forward_and_forget backend ctx ho;
Tensor.print ~with_code:false ~with_grad:false `Default @@ hey;
[%expect
- {|
- ┌─────────────────────────────────────────────────────────────┐
- │[0]: r2x4x3 shape 0:2|2:3->1:4 │
- │┌──────┬────────────────────────┬───────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 ││
- ││ │axis 2 │axis 2 ││
- │├──────┼────────────────────────┼───────────────────────────┤│
- ││axis 1│ 0.00 1.00 2.00 │ 1.20e+1 1.30e+1 1.40e+1 ││
- ││ │ 3.00 4.00 5.00 │ 1.50e+1 1.60e+1 1.70e+1 ││
- ││ │ 6.00 7.00 8.00 │ 1.80e+1 1.90e+1 2.00e+1 ││
- ││ │ 9.00 1.00e+1 1.10e+1 │ 2.10e+1 2.20e+1 2.30e+1 ││
- │└──────┴────────────────────────┴───────────────────────────┘│
- └─────────────────────────────────────────────────────────────┘
- |}];
+ {| [0]: r2x4x3 shape 0:2|2:3->1:4 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ ho;
[%expect
{|
@@ -361,28 +236,10 @@ let%expect_test "einsum outer product" =
Train.forward_and_forget backend ctx c;
Tensor.print ~with_code:false ~with_grad:false `Default @@ a;
[%expect
- {|
- ┌──────────────────┐
- │[0]: r2 shape 0:2 │
- │┌┬────────────┐ │
- │││axis 0 │ │
- │├┼────────────┤ │
- │││ 0.00 1.00 │ │
- │└┴────────────┘ │
- └──────────────────┘
- |}];
+ {| [0]: r2 shape 0:2 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ b;
[%expect
- {|
- ┌─────────────────────┐
- │[1]: r3 shape 0:3 │
- │┌┬──────────────────┐│
- │││axis 0 ││
- │├┼──────────────────┤│
- │││ 0.00 1.00 2.00 ││
- │└┴──────────────────┘│
- └─────────────────────┘
- |}];
+ {| [1]: r3 shape 0:3 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ c;
[%expect
{|
@@ -403,37 +260,10 @@ let%expect_test "einsum outer product" =
Train.forward_and_forget backend ctx c;
Tensor.print ~with_code:false ~with_grad:false `Default @@ a;
[%expect
- {|
- ┌─────────────────────────────────────────────────────────────┐
- │[5]: r2x4x3 shape 0:2|2:3->1:4 │
- │┌──────┬────────────────────────┬───────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 ││
- ││ │axis 2 │axis 2 ││
- │├──────┼────────────────────────┼───────────────────────────┤│
- ││axis 1│ 0.00 1.00 2.00 │ 1.20e+1 1.30e+1 1.40e+1 ││
- ││ │ 3.00 4.00 5.00 │ 1.50e+1 1.60e+1 1.70e+1 ││
- ││ │ 6.00 7.00 8.00 │ 1.80e+1 1.90e+1 2.00e+1 ││
- ││ │ 9.00 1.00e+1 1.10e+1 │ 2.10e+1 2.20e+1 2.30e+1 ││
- │└──────┴────────────────────────┴───────────────────────────┘│
- └─────────────────────────────────────────────────────────────┘
- |}];
+ {| [5]: r2x4x3 shape 0:2|2:3->1:4 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ b;
[%expect
- {|
- ┌──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[6]: r5x7x6 shape 0:5|2:6->1:7 │
- │┌──────┬─────────────────────────────────────────┬─────────────────────────────────────────┬─────────────────────────────────────────┬─────────────────────────────────────────┬─────────────────────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 │2 @ 0 │3 @ 0 │4 @ 0 ││
- ││ │axis 2 │axis 2 │axis 2 │axis 2 │axis 2 ││
- │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼─────────────────────────────────────────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│
- ││axis 1│ 0.00 1.00 ... 4.00 5.00 │ 4.20e+1 4.30e+1 ... 4.60e+1 4.70e+1 │ 8.40e+1 8.50e+1 ... 8.80e+1 8.90e+1 │ 1.26e+2 1.27e+2 ... 1.30e+2 1.31e+2 │ 1.68e+2 1.69e+2 ... 1.72e+2 1.73e+2 ││
- ││ │ 6.00 7.00 ... 1.00e+1 1.10e+1 │ 4.80e+1 4.90e+1 ... 5.20e+1 5.30e+1 │ 9.00e+1 9.10e+1 ... 9.40e+1 9.50e+1 │ 1.32e+2 1.33e+2 ... 1.36e+2 1.37e+2 │ 1.74e+2 1.75e+2 ... 1.78e+2 1.79e+2 ││
- ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││
- ││ │ 3.00e+1 3.10e+1 ... 3.40e+1 3.50e+1 │ 7.20e+1 7.30e+1 ... 7.60e+1 7.70e+1 │ 1.14e+2 1.15e+2 ... 1.18e+2 1.19e+2 │ 1.56e+2 1.57e+2 ... 1.60e+2 1.61e+2 │ 1.98e+2 1.99e+2 ... 2.02e+2 2.03e+2 ││
- ││ │ 3.60e+1 3.70e+1 ... 4.00e+1 4.10e+1 │ 7.80e+1 7.90e+1 ... 8.20e+1 8.30e+1 │ 1.20e+2 1.21e+2 ... 1.24e+2 1.25e+2 │ 1.62e+2 1.63e+2 ... 1.66e+2 1.67e+2 │ 2.04e+2 2.05e+2 ... 2.08e+2 2.09e+2 ││
- │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴─────────────────────────────────────────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│
- └──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ {| [6]: r5x7x6 shape 0:5|2:6->1:7 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ c;
[%expect
{|
@@ -612,93 +442,40 @@ let%expect_test "einsum matrix/inner+outer products" =
let%op f = a *+ "a|i->h; b|h->o => i->o" b in
Train.forward_and_forget backend ctx f;
Tensor.print ~with_code:false ~with_grad:false `Default @@ a2;
- [%expect
- {|
- ┌────────────────────────────────────────────────────────────────┐
- │[2]: ;=>_a2 shape 0:2|2:3->1:4 │
- │┌──────┬───────────────────────────┬───────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 ││
- ││ │axis 2 │axis 2 ││
- │├──────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 0.00 1.00 4.00 │ 1.44e+2 1.69e+2 1.96e+2 ││
- ││ │ 9.00 1.60e+1 2.50e+1 │ 2.25e+2 2.56e+2 2.89e+2 ││
- ││ │ 3.60e+1 4.90e+1 6.40e+1 │ 3.24e+2 3.61e+2 4.00e+2 ││
- ││ │ 8.10e+1 1.00e+2 1.21e+2 │ 4.41e+2 4.84e+2 5.29e+2 ││
- │└──────┴───────────────────────────┴───────────────────────────┘│
- └────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ c;
- [%expect
- {|
- ┌────────────────────────────────────────────────────────────────┐
- │[3]: ;=>_c shape 0:2|2:3->1:5 │
- │┌──────┬───────────────────────────┬───────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 ││
- ││ │axis 2 │axi2 ││
- │├──────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 4.20e+1 4.80e+1 5.40e+1 │ 1.43e+3 1.52e+3 1.60e+3 ││
- ││ │ 1.14e+2 1.36e+2 1.58e+2 │ 1.69e+3 1.80e+3 1.90e+3 ││
- ││ │ 1.86e+2 2.24e+2 2.62e+2 │ 1.96e+3 2.08e+3 2.19e+3 ││
- ││ │ 2.58e+2 3.12e+2 3.66e+2 │ 2.22e+3 2.36e+3 2.49e+3 ││
- ││ │ 3.30e+2 4.00e+2 4.70e+2 │ 2.49e+3 2.64e+3 2.79e+3 ││
- │└──────┴───────────────────────────┴───────────────────────────┘│
- └────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ d;
- [%expect
- {|
- ┌────────────────────────────────────────────────────────────────┐
- │[4]: ;=>_d shape 0:2,1:2|3:3->2:5 │
- │┌──────┬───────────────────────────┬───────────────────────────┐│
- ││ │0 @ 1 │1 @ 1 ││
- ││ │axis 3 │axis 3 ││
...TRUNCATED BY DUNE...
- ││ │0 @ 0 │1 @ 0 ││
- ││ │axis 2 │axis 2 ││
- │├──────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 1.80e+1 2.20e+1 2.60e+1 │ 6.60e+1 7.00e+1 7.40e+1 ││
- │└──────┴───────────────────────────┴───────────────────────────┘│
- └────────────────────────────────────────────────────────────────┘
- |}];
+ [%expect.unreachable];
let hey2 = TDSL.range_of_shape ~input_dims:[ 2 ] ~output_dims:[ 3 ] () in
let%op ho3 = hey2 ++ "...|...->... => 0" in
let ctx = Train.forward_and_ctx backend ctx ho3 in
Tensor.print ~with_code:false ~with_grad:false `Default @@ hey2;
- [%expect
- {|
- ┌─────────────────────────┐
- │[3]: r3x2 shape 1:2->0:3 │
- │┌──────┬────────────┐ │
- ││ │axis 1 │ │
- │├──────┼────────────┤ │
- ││axis 0│ 0.00 1.00 │ │
- ││ │ 2.00 3.00 │ │
- ││ │ 4.00 5.00 │ │
- │└──────┴────────────┘ │
- └─────────────────────────┘
- |}];
+ [%expect.unreachable];
Tensor.print ~with_code:false ~with_grad:false `Default @@ ho3;
- [%expect
- {|
- ┌──────────────────────┐
- │[4]: =>_ho3 shape 0:1 │
- │┌┬─────────┐ │
- │││axis 0 │ │
- │├┼─────────┤ │
- │││ 1.50e+1 │ │
- │└┴─────────┘ │
- └──────────────────────┘
- |}];
+ [%expect.unreachable];
let%op ho4 = hey2 ++ "i->j => i0j" in
Train.forward_and_forget backend ctx ho4;
Tensor.print ~with_code:false ~with_grad:false `Default @@ ho4;
- [%expect
- {|
- ┌──────────────────────────────┐
- │[5]: =>_ho4 shape 0:2,1:1,2:3 │
- │┌──────┬──────────────────┐ │
- ││ │axis 2 │ │
- │├──────┼──────────────────┤ │
- ││0 @ 0 │ 0.00 2.00 4.00 │ │
- ││axis 1│ │ │
- │├──────┼──────────────────┤ │
- ││1 @ 0 │ 1.00 3.00 5.00 │ │
- ││axis 1│ │ │
- │└──────┴──────────────────┘ │
- └──────────────────────────────┘
- |}]
+ [%expect.unreachable]
+[@@expect.uncaught_exn {|
+ (* CR expect_test_collector: This test expectation appears to contain a backtrace.
+ This is strongly discouraged as backtraces are fragile.
+ Please change this test to not include a backtrace. *)
+ (Invalid_argument
+ "Tnode.update_memory_mode: update 15 -> 35 inconsistent for r2x4x3")
+ Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45
+ Called from Ir__Low_level.visit_llc.(fun) in file "arrayjit/lib/low_level.ml", line 268, characters 13-53
+ Called from Base__Avltree.iter in file "src/avltree.ml", line 491, characters 4-16
+ Called from Base__Hashtbl.iteri in file "src/hashtbl.ml", line 282, characters 20-42
+ Re-raised at Base__Hashtbl.iteri in file "src/hashtbl.ml", line 288, characters 6-15
+ Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 875-876, characters 2-7
+ Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 869-886, characters 30-35
+ Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1149, characters 15-47
+ Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-75
+ Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-75
+ Called from Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 193-194, characters 4-45
+ Re-raised at Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 185-194, characters 34-47
+ Called from Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", line 356, characters 6-61
+ Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 355-356, characters 4-61
+ Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 354-362, characters 26-99
+ Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 494, characters 37-90
+ Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 488-497, characters 32-17
+ Called from Tutorials__Einsum_trivia.(fun) in file "test/einsum_trivia.ml", line 1277, characters 12-49
+ Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28
+ |}]
let%expect_test "einsum with fixed dim axes" =
Tensor.unsafe_reinitialize ();
@@ -1371,34 +742,10 @@ let%expect_test "einsum with fixed dim axes" =
Train.forward_and_forget backend ctx c;
Tensor.print ~with_code:false ~with_grad:false `Default @@ a;
[%expect
- {|
- ┌───────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[0]: r3x2x4 shape 0:3|2:4->1:2 │
- │┌──────┬────────────────────────┬────────────────────────────────────┬────────────────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 │2 @ 0 ││
- ││ │axis 2 │axis 2 │axis 2 ││
- │├──────┼────────────────────────┼────────────────────────────────────┼────────────────────────────────────┤│
- ││axis 1│ 0.00 1.00 2.00 3.00 │ 8.00 9.00 1.00e+1 1.10e+1 │ 1.60e+1 1.70e+1 1.80e+1 1.90e+1 ││
- ││ │ 4.00 5.00 6.00 7.00 │ 1.20e+1 1.30e+1 1.40e+1 1.50e+1 │ 2.00e+1 2.10e+1 2.20e+1 2.30e+1 ││
- │└──────┴────────────────────────┴────────────────────────────────────┴────────────────────────────────────┘│
- └───────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ {| [0]: r3x2x4 shape 0:3|2:4->1:2 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ b;
[%expect
- {|
- ┌────────────────────────────────┐
- │[1]: r3x4x1 shape 0:3|2:1->1:4 │
- │┌──────┬──────┬──────┬─────────┐│
- ││ │0 @ 0 │1 @ 0 │2 @ 0 ││
- ││ │axis 2│axis 2│axis 2 ││
- │├──────┼──────┼──────┼─────────┤│
- ││axis 1│ 0.00 │ 4.00 │ 8.00 ││
- ││ │ 1.00 │ 5.00 │ 9.00 ││
- ││ │ 2.00 │ 6.00 │ 1.00e+1 ││
- ││ │ 3.00 │ 7.00 │ 1.10e+1 ││
- │└──────┴──────┴──────┴─────────┘│
- └────────────────────────────────┘
- |}];
+ {| [1]: r3x4x1 shape 0:3|2:1->1:4 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ c;
[%expect
{|
@@ -1441,175 +788,175 @@ let%expect_test "outer_sum simulating axis concatenation" =
Tensor.print ~with_code:false ~with_grad:false `Default @@ positions;
[%expect
{|
- ┌────────────────────────────────┐
- │[9]: ;=>+ shape 0:4,1:5,2:6,3:3 │
- │┌──────┬──────────────────┐ │
- ││0 @ 0 │axis 3 │ │
- │├──────┼──────────────────┤ │
- ││0 @ 1 │ 0.00 0.00 0.00 │ │
- ││axis 2│ 0.00 0.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 0.00 0.00 4.00 │ │
- ││ │ 0.00 0.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││1 @ 1 │ 0.00 1.00 0.00 │ │
- ││axis 2│ 0.00 1.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 0.00 1.00 4.00 │ │
- ││ │ 0.00 1.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││2 @ 1 │ 0.00 2.00 0.00 │ │
- ││axis 2│ 0.00 2.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 0.00 2.00 4.00 │ │
- ││ │ 0.00 2.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││3 @ 1 │ 0.00 3.00 0.00 │ │
- ││axis 2│ 0.00 3.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 0.00 3.00 4.00 │ │
- ││ │ 0.00 3.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││4 @ 1 │ 0.00 4.00 0.00 │ │
- ││axis 2│ 0.00 4.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 0.00 4.00 4.00 │ │
- ││ │ 0.00 4.00 5.00 │ │
- │└──────┴──────────────────┘ │
- ├────────────────────────────────┤
- │┌──────┬──────────────────┐ │
- ││1 @ 0 │axis 3 │ │
- │├──────┼──────────────────┤ │
- ││0 @ 1 │ 1.00 0.00 0.00 │ │
- ││axis 2│ 1.00 0.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 1.00 0.00 4.00 │ │
- ││ │ 1.00 0.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││1 @ 1 │ 1.00 1.00 0.00 │ │
- ││axis 2│ 1.00 1.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 1.00 1.00 4.00 │ │
- ││ │ 1.00 1.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││2 @ 1 │ 1.00 2.00 0.00 │ │
- ││axis 2│ 1.00 2.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 1.00 2.00 4.00 │ │
- ││ │ 1.00 2.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││3 @ 1 │ 1.00 3.00 0.00 │ │
- ││axis 2│ 1.00 3.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 1.00 3.00 4.00 │ │
- ││ │ 1.00 3.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││4 @ 1 │ 1.00 4.00 0.00 │ │
- ││axis 2│ 1.00 4.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 1.00 4.00 4.00 │ │
- ││ │ 1.00 4.00 5.00 │ │
- │└──────┴──────────────────┘ │
- ├────────────────────────────────┤
- │┌──────┬──────────────────┐ │
- ││2 @ 0 │axis 3 │ │
- │├──────┼──────────────────┤ │
- ││0 @ 1 │ 2.00 0.00 0.00 │ │
- ││axis 2│ 2.00 0.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 2.00 0.00 4.00 │ │
- ││ │ 2.00 0.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││1 @ 1 │ 2.00 1.00 0.00 │ │
- ││axis 2│ 2.00 1.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 2.00 1.00 4.00 │ │
- ││ │ 2.00 1.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││2 @ 1 │ 2.00 2.00 0.00 │ │
- ││axis 2│ 2.00 2.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 2.00 2.00 4.00 │ │
- ││ │ 2.00 2.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││3 @ 1 │ 2.00 3.00 0.00 │ │
- ││axis 2│ 2.00 3.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 2.00 3.00 4.00 │ │
- ││ │ 2.00 3.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││4 @ 1 │ 2.00 4.00 0.00 │ │
- ││axis 2│ 2.00 4.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 2.00 4.00 4.00 │ │
- ││ │ 2.00 4.00 5.00 │ │
- │└──────┴──────────────────┘ │
- ├────────────────────────────────┤
- │┌──────┬──────────────────┐ │
- ││3 @ 0 │axis 3 │ │
- │├──────┼──────────────────┤ │
- ││0 @ 1 │ 3.00 0.00 0.00 │ │
- ││axis 2│ 3.00 0.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 3.00 0.00 4.00 │ │
- ││ │ 3.00 0.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││1 @ 1 │ 3.00 1.00 0.00 │ │
- ││axis 2│ 3.00 1.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 3.00 1.00 4.00 │ │
- ││ │ 3.00 1.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││2 @ 1 │ 3.00 2.00 0.00 │ │
- ││axis 2│ 3.00 2.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 3.00 2.00 4.00 │ │
- ││ │ 3.00 2.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││3 @ 1 │ 3.00 3.00 0.00 │ │
- ││axis 2│ 3.00 3.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 3.00 3.00 4.00 │ │
- ││ │ 3.00 3.00 5.00 │ │
- │├──────┼──────────────────┤ │
- ││4 @ 1 │ 3.00 4.00 0.00 │ │
- ││axis 2│ 3.00 4.00 1.00 │ │
- ││ │ ... ... ... │ │
- ││ │ 3.00 4.00 4.00 │ │
- ││ │ 3.00 4.00 5.00 │ │
- │└──────┴──────────────────┘ │
- └────────────────────────────────┘
+ ┌──────────────────────────────────────────┐
+ │[9]: ;=>+ shape 0:4,1:5,2:6,3:3 │
+ │┌──────┬─────────────────────────────────┐│
+ ││0 @ 0 │axis 3 ││
+ │├──────┼─────────────────────────────────┤│
+ ││0 @ 1 │ -1.13e+33 9.12e-41 -1.13e+33 ││
+ ││axis 2│ -5.65e+32 -1.16e+22 -1.13e+33 ││
+ ││ │ ... ... ... ││
+ ││ │ -5.65e+32 9.12e-41 -1.13e+33 ││
+ ││ │ -5.65e+32 4.56e-41 -1.13e+33 ││
+ │├──────┼─────────────────────────────────┤│
+ ││1 @ 1 │ -5.65e+32 1.00 -5.65e+32 ││
+ ││axis 2│ 9.12e-41 -1.16e+22 -5.65e+32 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 1.00 -5.65e+32 ││
+ ││ │ 4.56e-41 1.00 -5.65e+32 ││
+ │├──────┼─────────────────────────────────┤│
+ ││2 @ 1 │ -5.65e+32 2.00 -5.65e+32 ││
+ ││axis 2│ 4.56e-41 -1.16e+22 -5.65e+32 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 2.00 -5.65e+32 ││
+ ││ │ 5.60e-45 2.00 -5.65e+32 ││
+ │├──────┼─────────────────────────────────┤│
+ ││3 @ 1 │ -5.65e+32 3.00 -5.65e+32 ││
+ ││axis 2│ 4.56e-41 -1.16e+22 -5.65e+32 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 3.00 -5.65e+32 ││
+ ││ │ 0.00 3.00 -5.65e+32 ││
+ │├──────┼─────────────────────────────────┤│
+ ││4 @ 1 │ -5.65e+32 4.00 -5.65e+32 ││
+ ││axis 2│ 4.56e-41 -1.16e+22 -5.65e+32 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 4.00 -5.65e+32 ││
+ ││ │ 0.00 4.00 -5.65e+32 ││
+ │└──────┴─────────────────────────────────┘│
+ ├──────────────────────────────────────────┤
+ │┌──────┬─────────────────────────────────┐│
+ ││1 @ 0 │axis 3 ││
+ │├──────┼─────────────────────────────────┤│
+ ││0 @ 1 │ -1.13e+33 4.80e+1 -5.65e+32 ││
+ ││axis 2│ -5.65e+32 -1.16e+22 -5.65e+32 ││
+ ││ │ ... ... ... ││
+ ││ │ -5.65e+32 4.80e+1 -5.65e+32 ││
+ ││ │ -5.65e+32 4.80e+1 -5.65e+32 ││
+ │├──────┼─────────────────────────────────┤│
+ ││1 @ 1 │ -5.65e+32 4.90e+1 6.50e+1 ││
+ ││axis 2│ 1.00 -1.16e+22 6.60e+1 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 4.90e+1 6.90e+1 ││
+ ││ │ 1.00 4.90e+1 7.00e+1 ││
+ │├──────┼─────────────────────────────────┤│
+ ││2 @ 1 │ -5.65e+32 5.00e+1 6.50e+1 ││
+ ││axis 2│ 1.00 -1.16e+22 6.60e+1 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 5.00e+1 6.90e+1 ││
+ ││ │ 1.00 5.00e+1 7.00e+1 ││
+ │├──────┼─────────────────────────────────┤│
+ ││3 @ 1 │ -5.65e+32 5.10e+1 6.50e+1 ││
+ ││axis 2│ 1.00 -1.16e+22 6.60e+1 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 5.10e+1 6.90e+1 ││
+ ││ │ 1.00 5.10e+1 7.00e+1 ││
+ │├──────┼─────────────────────────────────┤│
+ ││4 @ 1 │ -5.65e+32 5.20e+1 6.50e+1 ││
+ ││axis 2│ 1.00 -1.16e+22 6.60e+1 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 5.20e+1 6.90e+1 ││
+ ││ │ 1.00 5.20e+1 7.00e+1 ││
+ │└──────┴─────────────────────────────────┘│
+ ├──────────────────────────────────────────┤
+ │┌──────┬─────────────────────────────────┐│
+ ││2 @ 0 │axis 3 ││
+ │├──────┼─────────────────────────────────┤│
+ ││0 @ 1 │ -1.13e+33 1.05e+2 -5.65e+32 ││
+ ││axis 2│ -5.65e+32 -1.16e+22 -5.65e+32 ││
+ ││ │ ... ... ... ││
+ ││ │ -5.65e+32 1.05e+2 -5.65e+32 ││
+ ││ │ -5.65e+32 1.05e+2 -5.65e+32 ││
+ │├──────┼─────────────────────────────────┤│
+ ││1 @ 1 │ -5.65e+32 1.06e+2 1.60e+2 ││
+ ││axis 2│ 2.00 -1.16e+22 1.61e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 1.06e+2 1.64e+2 ││
+ ││ │ 2.00 1.06e+2 1.65e+2 ││
+ │├──────┼─────────────────────────────────┤│
+ ││2 @ 1 │ -5.65e+32 1.07e+2 1.60e+2 ││
+ ││axis 2│ 2.00 -1.16e+22 1.61e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 1.07e+2 1.64e+2 ││
+ ││ │ 2.00 1.07e+2 1.65e+2 ││
+ │├──────┼─────────────────────────────────┤│
+ ││3 @ 1 │ -5.65e+32 1.08e+2 1.60e+2 ││
+ ││axis 2│ 2.00 -1.16e+22 1.61e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 1.08e+2 1.64e+2 ││
+ ││ │ 2.00 1.08e+2 1.65e+2 ││
+ │├──────┼─────────────────────────────────┤│
+ ││4 @ 1 │ -5.65e+32 1.09e+2 1.60e+2 ││
+ ││axis 2│ 2.00 -1.16e+22 1.61e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 1.09e+2 1.64e+2 ││
+ ││ │ 2.00 1.09e+2 1.65e+2 ││
+ │└──────┴─────────────────────────────────┘│
+ ├──────────────────────────────────────────┤
+ │┌──────┬─────────────────────────────────┐│
+ ││3 @ 0 │axis 3 ││
+ │├──────┼─────────────────────────────────┤│
+ ││0 @ 1 │ -1.13e+33 2.20e+2 -5.65e+32 ││
+ ││axis 2│ -5.65e+32 -1.16e+22 -5.65e+32 ││
+ ││ │ ... ... ... ││
+ ││ │ -5.65e+32 2.20e+2 -5.65e+32 ││
+ ││ │ -5.65e+32 2.20e+2 -5.65e+32 ││
+ │├──────┼─────────────────────────────────┤│
+ ││1 @ 1 │ -5.65e+32 2.21e+2 2.53e+2 ││
+ ││axis 2│ 3.00 -1.16e+22 2.54e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 2.21e+2 2.57e+2 ││
+ ││ │ 3.00 2.21e+2 2.58e+2 ││
+ │├──────┼─────────────────────────────────┤│
+ ││2 @ 1 │ -5.65e+32 2.22e+2 2.53e+2 ││
+ ││axis 2│ 3.00 -1.16e+22 2.54e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 2.22e+2 2.57e+2 ││
+ ││ │ 3.00 2.22e+2 2.58e+2 ││
+ │├──────┼─────────────────────────────────┤│
+ ││3 @ 1 │ -5.65e+32 2.23e+2 2.53e+2 ││
+ ││axis 2│ 3.00 -1.16e+22 2.54e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 2.23e+2 2.57e+2 ││
+ ││ │ 3.00 2.23e+2 2.58e+2 ││
+ │├──────┼─────────────────────────────────┤│
+ ││4 @ 1 │ -5.65e+32 2.24e+2 2.53e+2 ││
+ ││axis 2│ 3.00 -1.16e+22 2.54e+2 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 2.24e+2 2.57e+2 ││
+ ││ │ 3.00 2.24e+2 2.58e+2 ││
+ │└──────┴─────────────────────────────────┘│
+ └──────────────────────────────────────────┘
|}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ ti;
[%expect
{|
- ┌───────────────────────────┐
- │[1]: =>_ti shape 0:4,1:3 │
- │┌──────┬──────────────────┐│
- ││ │axis 1 ││
- │├──────┼──────────────────┤│
- ││axis 0│ 0.00 0.00 0.00 ││
- ││ │ 1.00 0.00 0.00 ││
- ││ │ 2.00 0.00 0.00 ││
- ││ │ 3.00 0.00 0.00 ││
- │└──────┴──────────────────┘│
- └───────────────────────────┘
+ ┌────────────────────────────────────┐
+ │[1]: =>_ti shape 0:4,1:3 │
+ │┌──────┬───────────────────────────┐│
+ ││ │axis 1 ││
+ │├──────┼───────────────────────────┤│
+ ││axis 0│ 0.00 4.56e-41 -5.65e+32 ││
+ ││ │ 1.00 4.80e+1 6.50e+1 ││
+ ││ │ 2.00 1.05e+2 1.60e+2 ││
+ ││ │ 3.00 2.20e+2 2.53e+2 ││
+ │└──────┴───────────────────────────┘│
+ └────────────────────────────────────┘
|}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ tk;
[%expect
{|
- ┌───────────────────────────┐
- │[7]: =>_tk shape 0:6,1:3 │
- │┌──────┬──────────────────┐│
- ││ │axis 1 ││
- │├──────┼──────────────────┤│
- ││axis 0│ 0.00 0.00 0.00 ││
- ││ │ 0.00 0.00 1.00 ││
- ││ │ ... ... ... ││
- ││ │ 0.00 0.00 4.00 ││
- ││ │ 0.00 0.00 5.00 ││
- │└──────┴──────────────────┘│
- └───────────────────────────┘
+ ┌─────────────────────────────────────┐
+ │[7]: =>_tk shape 0:6,1:3 │
+ │┌──────┬────────────────────────────┐│
+ ││ │axis 1 ││
+ │├──────┼────────────────────────────┤│
+ ││axis 0│ -5.65e+32 4.56e-41 0.00 ││
+ ││ │ 4.56e-41 -1.16e+22 1.00 ││
+ ││ │ ... ... ... ││
+ ││ │ -7.73e+22 4.56e-41 4.00 ││
+ ││ │ 0.00 1.40e-45 5.00 ││
+ │└──────┴────────────────────────────┘│
+ └─────────────────────────────────────┘
|}]
let%expect_test "einsum with a leftmost input axis preserved as output axis" =
@@ -1634,54 +981,10 @@ let%expect_test "einsum with a leftmost input axis preserved as output axis" =
Train.forward_and_forget backend ctx c;
Tensor.print ~with_code:false ~with_grad:false `Default @@ a;
[%expect
- {|
- ┌───────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[0]: r3x2x4_a shape 0:3|2:4->1:2 │
- │┌──────┬────────────────────────┬────────────────────────────────────┬────────────────────────────────────┐│
- ││ │0 @ 0 │1 @ 0 │2 @ 0 ││
- ││ │axis 2 │axis 2 │axis 2 ││
- │├──────┼────────────────────────┼────────────────────────────────────┼────────────────────────────────────┤│
- ││axis 1│ 0.00 1.00 2.00 3.00 │ 8.00 9.00 1.00e+1 1.10e+1 │ 1.60e+1 1.70e+1 1.80e+1 1.90e+1 ││
- ││ │ 4.00 5.00 6.00 7.00 │ 1.20e+1 1.30e+1 1.40e+1 1.50e+1 │ 2.00e+1 2.10e+1 2.20e+1 2.30e+1 ││
- │└──────┴────────────────────────┴────────────────────────────────────┴────────────────────────────────────┘│
- └───────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ {| [0]: r3x2x4_a shape 0:3|2:4->1:2 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ b;
[%expect
- {|
- ┌────────────────────────────────────────────────────────────────┐
- │[1]: r3x4x2x3_b shape 0:3|2:2,3:3->1:4 │
- │┌──────┬───────────────────────────┬───────────────────────────┐│
- ││0 @ 0 │0 @ 2 │1 @ 2 ││
- ││ │axis 3 │axis 3 ││
- │├──────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 0.00 1.00 2.00 │ 3.00 4.00 5.00 ││
- ││ │ 6.00 7.00 8.00 │ 9.00 1.00e+1 1.10e+1 ││
- ││ │ 1.20e+1 1.30e+1 1.40e+1 │ 1.50e+1 1.60e+1 1.70e+1 ││
- ││ │ 1.80e+1 1.90e+1 2.00e+1 │ 2.10e+1 2.20e+1 2.30e+1 ││
- │└──────┴───────────────────────────┴───────────────────────────┘│
- ├────────────────────────────────────────────────────────────────┤
- │┌──────┬───────────────────────────┬───────────────────────────┐│
- ││1 @ 0 │0 @ 2 │1 @ 2 ││
- ││ │axis 3 │axis 3 ││
- │├──────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 2.40e+1 2.50e+1 2.60e+1 │ 2.70e+1 2.80e+1 2.90e+1 ││
- ││ │ 3.00e+1 3.10e+1 3.20e+1 │ 3.30e+1 3.40e+1 3.50e+1 ││
- ││ │ 3.60e+1 3.70e+1 3.80e+1 │ 3.90e+1 4.00e+1 4.10e+1 ││
- ││ │ 4.20e+1 4.30e+1 4.40e+1 │ 4.50e+1 4.60e+1 4.70e+1 ││
- │└──────┴───────────────────────────┴───────────────────────────┘│
- ├────────────────────────────────────────────────────────────────┤
- │┌──────┬───────────────────────────┬───────────────────────────┐│
- ││2 @ 0 │0 @ 2 │1 @ 2 ││
- ││ │axis 3 │axis 3 ││
- │├──────┼───────────────────────────┼───────────────────────────┤│
- ││axis 1│ 4.80e+1 4.90e+1 5.00e+1 │ 5.10e+1 5.20e+1 5.30e+1 ││
- ││ │ 5.40e+1 5.50e+1 5.60e+1 │ 5.70e+1 5.80e+1 5.90e+1 ││
- ││ │ 6.00e+1 6.10e+1 6.20e+1 │ 6.30e+1 6.40e+1 6.50e+1 ││
- ││ │ 6.60e+1 6.70e+1 6.80e+1 │ 6.90e+1 7.00e+1 7.10e+1 ││
- │└──────┴───────────────────────────┴───────────────────────────┘│
- └────────────────────────────────────────────────────────────────┘
- |}];
+ {| [1]: r3x4x2x3_b shape 0:3|2:2,3:3->1:4 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ c;
[%expect
{|
@@ -1717,34 +1020,10 @@ let%expect_test "einsum permuting two leftmost input axes as output axes" =
Train.forward_and_forget backend ctx c;
Tensor.print ~with_code:false ~with_grad:false `Default @@ a;
[%expect
- {|
- ┌───────────────────────────┐
- │[0]: r2x2_a shape 1:2->0:2 │
- │┌──────┬────────────┐ │
- ││ │axis 1 │ │
- │├──────┼────────────┤ │
- ││axis 0│ 0.00 1.00 │ │
- ││ │ 2.00 3.00 │ │
- │└──────┴────────────┘ │
- └───────────────────────────┘
- |}];
+ {| [0]: r2x2_a shape 1:2->0:2 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ b;
[%expect
- {|
- ┌───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐
- │[1]: r2x2x3x4_b shape 1:2,2:3,3:4->0:2 │
- │┌──────┬────────────────────────────────────┬────────────────────────────────────┬────────────────────────────────────┐│
- ││ │0 @ 2 │1 @ 2 │2 @ 2 ││
- ││ │axis 3 │axis 3 │axis 3 ││
- │├──────┼────────────────────────────────────┼────────────────────────────────────┼────────────────────────────────────┤│
- ││0 @ 1 │ 0.00 1.00 2.00 3.00 │ 4.00 5.00 6.00 7.00 │ 8.00 9.00 1.00e+1 1.10e+1 ││
- ││axis 0│ 2.40e+1 2.50e+1 2.60e+1 2.70e+1 │ 2.80e+1 2.90e+1 3.00e+1 3.10e+1 │ 3.20e+1 3.30e+1 3.40e+1 3.50e+1 ││
- │├──────┼────────────────────────────────────┼────────────────────────────────────┼────────────────────────────────────┤│
- ││1 @ 1 │ 1.20e+1 1.30e+1 1.40e+1 1.50e+1 │ 1.60e+1 1.70e+1 1.80e+1 1.90e+1 │ 2.00e+1 2.10e+1 2.20e+1 2.30e+1 ││
- ││axis 0│ 3.60e+1 3.70e+1 3.80e+1 3.90e+1 │ 4.00e+1 4.10e+1 4.20e+1 4.30e+1 │ 4.40e+1 4.50e+1 4.60e+1 4.70e+1 ││
- │└──────┴────────────────────────────────────┴────────────────────────────────────┴────────────────────────────────────┘│
- └───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘
- |}];
+ {| [1]: r2x2x3x4_b shape 1:2,2:3,3:4->0:2 <virtual> |}];
Tensor.print ~with_code:false ~with_grad:false `Default @@ c;
[%expect
{|
"/usr/bin/env" "bash" "-c" "opam exec -- dune build @install @check @runtest && rm -rf _build" failed with exit status 1
2025-06-27 14:02.53: Job failed: Failed: Build failed