2025-06-28 12:16.25: New job: test ahrefs/ocannl https://github.com/ahrefs/ocannl.git#refs/heads/master (65255f6690d4e579cc2738467d468f941d5aac3a) (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 65255f66 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-28 12:16.25: Using cache hint "ahrefs/ocannl-ocaml/opam:fedora-42-ocaml-5.3@sha256:ca935fbf20f8d83e78dff4d973011a91d32565d7d98af772da42600dd103623c-fedora-42-5.3_opam-2.3-82912e70cd030e6e6af979ac4425a6a2" 2025-06-28 12:16.25: 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-28 12:16.25: Waiting for resource in pool OCluster 2025-06-28 12:16.25: Waiting for worker… 2025-06-28 12:25.06: Got resource from pool OCluster Building on toxis.caelum.ci.dev All commits already cached HEAD is now at 65255f66 Propagate optimization context (with the computations table for Low_level optimizer) (from ocaml/opam:fedora-42-ocaml-5.3@sha256:ca935fbf20f8d83e78dff4d973011a91d32565d7d98af772da42600dd103623c) 2025-06-28 12:25.11 ---> 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-28 12:25.11 ---> using "b1f0db1fada5631e0e8a089e4b21d3aba9fcebb8766eb6c5a9e6a096ca963ec7" from cache /src: (run (shell "sudo ln -f /usr/bin/opam-2.3 /usr/bin/opam")) 2025-06-28 12:25.11 ---> 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-28 12:25.11 ---> 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-28 12:25.11 ---> using "626b7abede693a100c736148e8f886bf78f49cecf8d70aa67da157cd49d141e7" from cache /src: (workdir /src) /src: (run (shell "sudo chown opam /src")) 2025-06-28 12:25.11 ---> 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-28 12:25.11 ---> using "3a2c7f3d7f858cbf5495fa9633d9d7ab56096a2d0bdf3152c92a7cc6bef114ec" from cache /src: (copy (src neural_nets_lib.opam arrayjit.opam) (dst ./)) 2025-06-28 12:25.11 ---> 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-28 12:25.11 ---> using "d8bd79ed7ec398484a3126847d77f7404e5dc6df25da16ff537d70577bdcbf99" from cache /src: (run (network host) (shell "echo '(lang dune 3.0)' > './dune-project'")) 2025-06-28 12:25.11 ---> 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-28 12:25.11 ---> 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-28 12:25.11 ---> using "22f4aa9826c93953a9d195932d45b7588c483aa23d15a0dcfeeed4b6ac79d440" from cache /src: (copy (src .) (dst /src)) 2025-06-28 12:25.12 ---> saved as "9b074ca898eca75f3d199e190eb88945cb7f464114f998a3159a328ab025d2b8" /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 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/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: (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition 'Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/fce3bc453d8e67b38d947e91b64d1c5c/default/test/ocannl_config.' -source-tree-root .. -diff-cmd -) Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file File "test/einsum/dune", line 5, characters 7-25: 5 | (name moons_demo_variant) ^^^^^^^^^^^^^^^^^^ (cd _build/default/test/einsum && ./moons_demo_variant.exe) > _build/default/test/einsum/moons_demo_variant.exe.output Fatal error: exception Invalid_argument("Tnode.update_memory_mode: update 40 -> 152 inconsistent for n43") Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 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 378, characters 6-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 Called from Dune__exe__Moons_demo_variant in file "test/einsum/moons_demo_variant.ml", line 62, characters 4-83 (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/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)))) @!Fatal error: exception Invalid_argument("Tnode.update_memory_mode: update 40 -> 152 inconsistent for n40") Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 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 378, characters 6-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 Called from Ocannl__Train.example_train_loop in file "lib/train.ml", line 408, characters 20-89 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 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 40 -> 152 inconsistent for n2") Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 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 378, characters 6-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 Called from Dune__exe__Einsum_trivia_exec in file "test/einsum/einsum_trivia_exec.ml", line 94, characters 2-40 (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/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 Fatal error: exception Invalid_argument("Tnode.update_memory_mode: update 40 -> 152 inconsistent for n21") Raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 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 378, characters 6-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 Called from Dune__exe__Micrograd_demo_logging in file "test/micrograd_demo_logging.ml", line 34, characters 13-67 (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file 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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/zero2hero_1of7.ml.corrected diff --git a/_build/default/test/zero2hero_1of7.ml b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/zero2hero_1of7.ml.corrected index 61f11e7..4c8d05e 100644 --- a/_build/default/test/zero2hero_1of7.ml +++ b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/zero2hero_1of7.ml.corrected @@ -28,54 +28,14 @@ let%expect_test "Graph drawing recompile" = Train.set_hosted x.value; Train.forward_and_forget backend ctx f_nd; Tensor.print_tree ~with_grad:true ~depth:9 f_nd; - [%expect - {| - #15 +_f_nd - 6.00e+1 - #16 grad_+_f_nd Virt/30 - <void> - #13 - Virt/152 │#2 5. Virt/40 - <void> │<void> - #14 grad_- Virt/30 │ - <void> │ - #11 *. Virt/152 │ #4 *. Virt/152 │ - <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 │ - <void> │<void> │<void> │ 5.00 │ - │#8 grad_**. Virt/30│ │#1 grad_x Local/30│ - │<void> │ │<void> │ - │[0]│ #6 2. Virt/40 │ │ │ - │ │ <void> │ │ │ - |}]; + [%expect.unreachable]; let%op f = (3 *. ("x" [ 5 ] **. 2)) - (4 *. x) + 5 in Train.every_non_literal_on_host f; let f_upd = Train.grad_update f in let f_bprop = Train.to_routine (module Backend) ctx IDX.empty f_upd in Train.run f_bprop; Tensor.print_tree ~with_grad:true ~depth:9 f; - [%expect - {| - #32 +_f - 6.00e+1 - #33 grad_+_f - 1.00 - #30 - │#19 5. Virt/40 - 5.50e+1 │<void> - #31 grad_- │ - 1.00 │ - #28 *. │ #21 *. │ - 7.50e+1 │ 2.00e+1 │ - #29 grad_*. │ #22 grad_*. │ - 1.00 │ -1.00 │ - #27 3. Virt/40│ #24 **. │#20 4. Virt/40│#17 x │ - <void> │ 2.50e+1 │<void> │ 5.00 │ - │ #25 grad_**. │ │#18 grad_x│ - │ 3.00 │ │ 2.60e+1 │ - │[17]│#23 2. Virt/40│ │ │ - │ │<void> │ │ │ - |}]; + [%expect.unreachable]; let xs = Array.init 10 ~f:Float.(fun i -> of_int i - 5.) in let ys = Array.map xs ~f:(fun v -> @@ -94,54 +54,36 @@ let%expect_test "Graph drawing recompile" = [ Scatterplot { points = Array.zip_exn xs ys; content = PrintBox.line "#" } ] in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 1.00e+2│# │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │f │ │ - │( │ │ - │x │ │ - │) │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ #│ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ # │ - │ │ # │ - │ │ │ - │ 4.00 │ # # │ - ├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │-5.00 4.00│ - │ │ x │ - └────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 152 inconsistent for n2") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 + Called from Tutorials__Zero2hero_1of7.(fun) in file "test/zero2hero_1of7.ml", line 29, characters 2-43 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "Graph drawing fetch" = Tensor.unsafe_reinitialize (); @@ -156,19 +98,7 @@ let%expect_test "Graph drawing fetch" = Train.every_non_literal_on_host f5; Train.forward_and_forget (module Backend) ctx f5; Tensor.print_tree ~with_grad:false ~depth:9 f5; - [%expect - {| - #9 +_f_5. - 6.00e+1 - #8 - │#1 5. Virt/40 - 5.50e+1 │<void> - #7 *. │ #3 *. │ - 7.50e+1 │ 2.00e+1 │ - #6 3. Virt/40│ #5 **. │#2 4. Virt/40│#0 5. Virt/40│ - <void> │ 2.50e+1 │<void> │<void> │ - │[0]│#4 2. Virt/40│ │ │ - │ │<void> │ │ │ - |}]; + [%expect.unreachable]; let size = 100 in let xs = Array.init size ~f:Float.(fun i -> (of_int i / 10.) - 5.) in (* Yay, the whole shape gets inferred! *) @@ -202,54 +132,36 @@ let%expect_test "Graph drawing fetch" = ] in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 1.00e+2 │# │ - │ │# │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ # │ - │ │ # # #│ - │ │ # # # │ - │ │ # # │ - │ │ # ## │ - │ │ ## # │ - │f │ # # │ - │( │ # # ## │ - │x │ ## # │ - │) │ # # │ - │ │ # # ### **│ - │ │ ## # * **** │ - │ │ # # ## * **** │ - │ │ ### ## * *** │ - │ │ # # # ##** ** │ - │ │ #### # ###* * │ - │ │ # #### # # ## # #### │ - │ │ # # ** * │ - │ │- - - - - - - - - - - * ** * - - - - - - - - │ - │ │ * **** │ - │ │ * **** │ - │ │ **** * │ - │ │ * **** │ - │ │ * **** │ - │ │ * **** │ - │ │ ** *** │ - │ │ * * *** │ - │ -3.40e+1│* * ** │ - ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │-5.00 4.90│ - │ │ x │ - └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 152 inconsistent for n1") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 + Called from Tutorials__Zero2hero_1of7.(fun) in file "test/zero2hero_1of7.ml", line 157, characters 2-50 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "Simple gradients hosted" = Tensor.unsafe_reinitialize (); @@ -271,96 +183,51 @@ let%expect_test "Simple gradients hosted" = let sgd_routine = Train.to_routine (module Backend) grad_routine.context IDX.empty sgd in (* Check out the initial state without running a forward pass. *) Tensor.print_tree ~spy:true ~with_grad:true ~depth:9 l; - [%expect - {| - #12 *._l Host&stream/41 - <not-in-yet> - #13 grad_*._l Host&stream/41 - <not-in-yet> - #8 +_d Host&stream/41 │#10 f Host&shared/39 - <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 │ - <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 │ │ - <not-in-yet> │<not-in-yet> │ │ - #1 grad_a Host&stream/41│#3 grad_b Host&stream/41│ │ - <not-in-yet> │<not-in-yet> │ │ - |}]; + [%expect.unreachable]; (* Do not update the params: all values and gradients will be at initial points, which are specified in the tensor in the brackets. *) Train.run grad_routine; Tensor.print_tree ~with_grad:true ~depth:9 l; - [%expect - {| - #12 *._l - -8.00 - #13 grad_*._l - 1.00 - #8 +_d │#10 f - 4.00 │ -2.00 - #9 grad_+_d │#11 grad_f - -2.00 │ 4.00 - #4 *._e │#6 c │ - -6.00 │ 1.00e+1 │ - #5 grad_*._e │#7 grad_c│ - -2.00 │ -2.00 │ - #0 a │#2 b │ │ - 2.00 │ -3.00 │ │ - #1 grad_a│#3 grad_b│ │ - 6.00 │ -4.00 │ │ - |}]; + [%expect.unreachable]; (* Now we update the params, but we are not doing the forward and backward passes: only params values will change, compared to the above. The update is in the opposite direction of the gradient. *) Train.run sgd_routine; Tensor.print_tree ~with_grad:true ~depth:9 l; - [%expect - {| - #12 *._l - -8.00 - #13 grad_*._l - 1.00 - #8 +_d │#10 f - 4.00 │ -2.40 - #9 grad_+_d │#11 grad_f - -2.00 │ 4.00 - #4 *._e │#6 c │ - -6.00 │ 1.02e+1 │ - #5 grad_*._e │#7 grad_c│ - -2.00 │ -2.00 │ - #0 a │#2 b │ │ - 1.40 │ -2.60 │ │ - #1 grad_a│#3 grad_b│ │ - 6.00 │ -4.00 │ │ - |}]; + [%expect.unreachable]; (* 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. *) Train.run grad_routine; Tensor.print_tree ~with_grad:true ~depth:9 l; - [%expect - {| - #12 *._l - -1.57e+1 - #13 grad_*._l - 1.00 - #8 +_d │#10 f - 6.56 │ -2.40 - #9 grad_+_d │#11 grad_f - -2.40 │ 6.56 - #4 *._e │#6 c │ - -3.64 │ 1.02e+1 │ - #5 grad_*._e │#7 grad_c│ - -2.40 │ -2.40 │ - #0 a │#2 b │ │ - 1.40 │ -2.60 │ │ - #1 grad_a│#3 grad_b│ │ - 6.24 │ -3.36 │ │ - |}] + [%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 40 -> 152 inconsistent for n14") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Zero2hero_1of7.(fun) in file "test/zero2hero_1of7.ml", line 270, characters 21-73 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "Simple gradients virtual" = Tensor.unsafe_reinitialize (); @@ -387,62 +254,26 @@ 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 (* Check out the state without running a forward pass or compiling the SGD update. *) 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> │ │ - |}]; + [%expect.unreachable]; (* Do not update the params: all values and gradients will be at initial points, which are specified in the tensor in the brackets. *) Train.run grad_routine; 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> │ │ - |}]; + [%expect.unreachable]; (* Only now compile the SGD update. *) let sgd_routine = Train.to_routine (module Backend) grad_routine.context IDX.empty sgd in (* Now we update the params, but are not doing the forward and backward passes: only params values @@ -450,48 +281,39 @@ let%expect_test "Simple gradients virtual" = always be recomputed using the latest parameter state. *) Train.run sgd_routine; 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> │ │ - |}]; + [%expect.unreachable]; (* 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. *) Train.run grad_routine; 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> │ │ - |}] + [%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 40 -> 152 inconsistent for e") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Zero2hero_1of7.(fun) in file "test/zero2hero_1of7.ml", line 401, characters 21-73 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "tanh plot" = Tensor.unsafe_reinitialize (); @@ -510,21 +332,34 @@ let%expect_test "2D neuron hosted" = let routine = Train.to_routine (module Backend) ctx IDX.empty update in Train.run routine; Tensor.print_tree ~with_grad:true ~depth:9 v; - [%expect - {| - #8 +_v - 7.00e-1 - #9 grad_+_v - 1.00 - #6 * │#0 b - -6.00 │ 6.70 - #7 grad_* │#1 grad_b - 1.00 │ 1.00 - #2 w │#4 x │ - -3.00 1.00 │ 2.00 0.00 │ - #3 grad_w │#5 grad_x │ - 2.00 0.00 │ -3.00 1.00 │ - |}] + [%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 40 -> 152 inconsistent for n10") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Zero2hero_1of7.(fun) in file "test/zero2hero_1of7.ml", line 510, characters 16-70 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "2D neuron virtual" = Tensor.unsafe_reinitialize (); @@ -537,18 +372,31 @@ let%expect_test "2D neuron virtual" = let routine = Train.to_routine (module Backend) ctx IDX.empty update in Train.run routine; Tensor.print_tree ~with_grad:true ~depth:9 v; - [%expect - {| - #8 +_v - 7.00e-1 - #9 grad_+_v Virt/40 - <void> - #6 * Local/46 │#0 b - <void> │ 6.70 - #7 grad_* Virt/40 │#1 grad_b Local/46 - <void> │<void> - #2 w │#4 x │ - -3.00 1.00 │ 2.00 0.00 │ - #3 grad_w Local/46│#5 grad_x Local/46│ - <void> │<void> │ - |}] + [%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 40 -> 151 inconsistent for n6.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Zero2hero_1of7.(fun) in file "test/zero2hero_1of7.ml", line 537, characters 16-70 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] File "test/primitive_ops.ml", line 1, characters 0-0: /usr/sbin/git --no-pager diff --no-index --color=always -u _build/default/test/primitive_ops.ml _build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/primitive_ops.ml.corrected diff --git a/_build/default/test/primitive_ops.ml b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/primitive_ops.ml.corrected index 17afc2d..d095e04 100644 --- a/_build/default/test/primitive_ops.ml +++ b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/primitive_ops.ml.corrected @@ -53,740 +53,474 @@ let%expect_test "relu" = let%op f x = relu x in let plot_box = plot_unop ~f () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 5.00│ #│ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ # │ - │f │ # │ - │( │ ## │ - │x │ # │ - │) │ # │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ * ******************************* **** ******* ****│ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ ## │ - │ 0.00│**** *** *** *** ********-*************** ********- - - - - - - - - - │ - ├─────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │-5.00 5.00│ - │ │ x │ - └─────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for relu_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 54, characters 17-32 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "sat01" = let%op f x = sat01 x in let plot_box = plot_unop ~f () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 1.00│ * *********###################### #### ####### ####│ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │f │ # │ - │( │ │ - │x │ │ - │) │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ │ │ - │ │ │ - │ │ │ - │ │ # │ - │ 0.00│**** *** *** *** ********-*************** ********- - ********************** **** *******-****│ - ├─────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │-5.00 5.00│ - │ │ x │ - └─────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for sat01_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 107, characters 17-32 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "exp(x)" = let%op f x = exp x in let plot_box = plot_unop ~f ~x_max:1.0 () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 2.71│ *│ - │ │ │ - │ │ │ - │ │ * │ - │ │ │ - │ │ * │ - │ │ │ - │ │ * │ - │ │ │ - │ │ * │ - │ │ │ - │ │ * │ - │ │ * │ - │ │ │ - │ │ * │ - │ │ * │ - │ │ │ - │ │ * │ - │f │ * │ - │( │ * │ - │x │ │ - │) │ * │ - │ │ * │ - │ │ * │ - │ │ * │ - │ │ * │ - │ │ ** │ - │ │ * │ - │ │ * │ - │ │ ** │ - │ │ ** │ - │ │ ** │ - │ │ * │ - │ │ ** │ - │ │ *** │ - │ │ **** │ - │ │ ***** │ - │ │ ******* │ - │ │ *********** │ - │ 0.00│********* ************ ************** * - - - - - - - - - - - - │ - ├─────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │-5.00 1.00│ - │ │ x │ - └─────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for exp_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 160, characters 17-43 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "log(x)" = let%op f x = log x in let plot_box = plot_unop ~f ~x_min:0.1 ~x_max:5.0 () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 1.00e+1│* │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ * │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ * │ - │ │ │ - │f │ │ - │( │ * │ - │x │ │ - │) │ │ - │ │ * │ - │ │ * │ - │ │ * │ - │ │ * │ - │ │ ** │ - │ │ ** #############│ - │ │ *** ################### #### │ - │ │ ****** ######### #### ## │ - │ │ ******** *****##### #### │ - │ │ ###### ## ***** ************* **** ********************* ******* │ - │ │- - - - ####### - - - - - - - - - - - - - **********│ - │ │ ### # │ - │ │ #### │ - │ │ ## │ - │ │ ## │ - │ │ ## │ - │ │ # │ - │ -2.30 │# │ - ├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │1.00e-1 5.00│ - │ │ x │ - └────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for log_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 213, characters 17-54 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "log2(x)" = let%op f x = log2 x in let plot_box = plot_unop ~f ~x_min:0.1 ~x_max:5.0 () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 1.44e+1│* │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ * │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ * │ - │ │ │ - │f │ │ - │( │ * │ - │x │ │ - │) │ │ - │ │ * │ - │ │ * │ - │ │ * │ - │ │ * │ - │ │ ** │ - │ │ ** #############│ - │ │ *** ################### #### │ - │ │ ****** ######### #### ## │ - │ │ ******** *****##### #### │ - │ │ ###### ## ***** ************* **** ********************* ******* │ - │ │- - - - ####### - - - - - - - - - - - - - **********│ - │ │ ### # │ - │ │ #### │ - │ │ ## │ - │ │ ## │ - │ │ ## │ - │ │ # │ - │ -3.32 │# │ - ├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │1.00e-1 5.00│ - │ │ x │ - └────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for log2_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 266, characters 17-54 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "sin(x)" = let%op f x = sin x in let plot_box = plot_unop ~f () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 9.99e-1 │ # │ - │ │#### ## *** ** ### ### │ ...TRUNCATED BY DUNE... + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 478, characters 17-32 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "sqrt(x)" = let%op f x = sqrt x in let plot_box = plot_unop ~f ~x_min:0.1 ~x_max:5.0 () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 2.23│ #│ - │ │ ##### │ - │ │ ##### │ - │ │ #### │ - │ │ ## ## │ - │ │ ##### │ - │ │ #### │ - │ │ ##### │ - │ │ #### │ - │ │ ## # │ - │ │ # ## │ - │ │ ### │ - │ │* #### │ - │ │ #### │ - │ │ # # │ - │ │ ### │ - │ │ ### │ - │ │ * ### │ - │f │ ## │ - │( │ ### │ - │x │ * ### │ - │) │ ## │ - │ │ * ## │ - │ │ # ## │ - │ │ * ## │ - │ │ * ## │ - │ │ ** # │ - │ │ *# │ - │ │ # ** │ - │ │ ## *** │ - │ │ # * * │ - │ │ # ***** │ - │ │ # ******** │ - │ │ # * ********** │ - │ │# * ************* **** │ - │ │ * ********************* ************ │ - │ │ *****│ - │ │ │ - │ │ │ - │ 0.00│- - - - - - - - - - - - - - - - - - - - │ - ├─────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │1.00e-1 5.00│ - │ │ x │ - └─────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for sqrt_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 531, characters 17-54 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "recip(x)" = let%op f x = recip x in let plot_box = plot_unop ~f ~x_min:0.1 ~x_max:5.0 () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 1.00e+1 │# │ - │ │ │ - │ │ ## │ - │ │ ######### │ - │ │- - - ###-**************-********** *************-****-********************* *****************│ - │ │ ******* │ - │ │ ** │ - │ │ * │ - │ │ * │ - │ │ │ - │ │ * │ - │ │ │ - │ │ │ - │ │ * │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │f │ │ - │( │ │ - │x │ * │ - │) │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ -1.00e+2│* │ - ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │1.00e-1 5.00│ - │ │ x │ - └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for recip_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 584, characters 17-54 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "recip_sqrt(x)" = let%op f x = recip_sqrt x in let plot_box = plot_unop ~f ~x_min:0.1 ~x_max:5.0 () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 3.16 │# │ - │ │ │ - │ │ ## │ - │ │ ## │ - │ │ ####### │ - │ │ ### ############## ##### │ - │ │ ##### ############# #### ##################### #################│ - │ │- - - - - - -********** *************-****-********************* *****************│ - │ │ * ************** │ - │ │ **** │ - │ │ *** │ - │ │ * │ - │ │ * │ - │ │ * │ - │ │ │ - │ │ * │ - │ │ │ - │ │ │ - │f │ │ - │( │ * │ - │x │ │ - │) │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ * │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ -1.58e+1│* │ - ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │1.00e-1 5.00│ - │ │ x │ - └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for recip_sqrt_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 637, characters 17-54 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "tanh(x)" = let%op f x = tanh x in let plot_box = plot_unop ~f () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 9.99e-1 │ #│ - │ │ ** * ############## #### ####### ### │ - │ │ * * #### │ - │ │ * * ## │ - │ │ * * # │ - │ │ * * ## │ - │ │ # │ - │ │ * * │ - │ │ * * │ - │ │ # │ - │ │ * # * │ - │ │ * * │ - │ │ # │ - │ │ * # * │ - │ │ * * │ - │ │ * # * │ - │ │ ** ** │ - │ │ ** # ** │ - │f │ ** ** │ - │( │ * ***** # ****** │ - │x │**** *** *** *** ********- - - - - - - - - - ******* **** *******-****│ - │) │ # │ - │ │ │ - │ │ # │ - │ │ │ - │ │ # │ - │ │ │ - │ │ # │ - │ │ # │ - │ │ │ - │ │ # │ - │ │ # │ - │ │ # │ - │ │ │ - │ │ # │ - │ │ ## │ - │ │ # │ - │ │ ## │ - │ │ #### │ - │ -9.99e-1│#### ### ### ### ######## ###### │ - ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │-5.00 5.00│ - │ │ x │ - └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for tanh_f_x.grad") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 690, characters 17-32 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "where(x < 0, sin(x), cos(x))" = let%op f x = where (x < !.0.) (sin x) (cos x) in let plot_box = plot_unop ~f () in PrintBox_text.output Stdio.stdout plot_box; - [%expect - {| - ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 9.99e-1 │ # * │ - │ │#### ## *** ## ** ***│ - │ │ # ** ## * │ - │ │ # * # * │ - │ │ # * │ - │ │ # * # * │ - │ │ * # │ - │ │ # * # * │ - │ │ # * │ - │ │ * # │ - │ │ # * # * │ - │ │ │ - │ │ # * # * │ - │ │ # * │ - │ │* * # #│ - │ │ # * │ - │ │ * * # # │ - │ │ # * │ - │f │ * * # # │ - │( │ # * │ - │x │- * - - - - - - *- - - - - - # - - - - - -# │ - │) │ # * * │ - │ │ * * # # │ - │ │ # # * * │ - │ │ * * # # │ - │ │ # # * * │ - │ │ * * # # │ - │ │ # # * * │ - │ │ * * # # │ - │ │ # * # * # * │ - │ │ * # # * * # │ - │ │ * * # # │ - │ │ # # * * │ - │ │ * * # * # # │ - │ │ * * * # │ - │ │ * # # * * # │ - │ │ * * # # * * # # │ - │ │ * * # # * * # # │ - │ │ ** * # # * * # ## │ - │ -9.99e-1│ * ***** ####### ******* #### # │ - ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │-5.00 5.00│ - │ │ x │ - └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 151 inconsistent for n12") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 616, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Primitive_ops.plot_unop in file "test/primitive_ops.ml", line 35, characters 19-72 + Called from Tutorials__Primitive_ops.(fun) in file "test/primitive_ops.ml", line 743, characters 17-32 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] 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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/moons_demo_parallel.ml.corrected diff --git a/_build/default/test/moons_demo_parallel.ml b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/moons_demo_parallel.ml.corrected index ce09725..5dffa84 100644 --- a/_build/default/test/moons_demo_parallel.ml +++ b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/moons_demo_parallel.ml.corrected @@ -105,4 +105,31 @@ 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. *) + (Invalid_argument + "Tnode.update_memory_mode: update 40 -> 152 inconsistent for n39") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.example_train_loop in file "lib/train.ml", line 408, characters 20-89 + 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/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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/micrograd_demo.ml.corrected diff --git a/_build/default/test/micrograd_demo.ml b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/micrograd_demo.ml.corrected index 59a566d..0858a34 100644 --- a/_build/default/test/micrograd_demo.ml +++ b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/micrograd_demo.ml.corrected @@ -31,57 +31,38 @@ let%expect_test "Micrograd README basic example" = let step = Train.to_routine (module Backend) ctx IDX.empty update in Train.run step; Tensor.print ~with_code:false ~with_grad:false `Default g; - [%expect - {| - ┌────────────────────┐ - │[75]: +_g shape 0:1 │ - │┌┬─────────┐ │ - │││axis 0 │ │ - │├┼─────────┤ │ - │││ 2.47e+1 │ │ - │└┴─────────┘ │ - └────────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:true `Default a; - [%expect - {| - ┌─────────────────┐ - │[0]: a shape 0:1 │ - │┌┬───────┐ │ - │││axis 0 │ │ - │├┼───────┤ │ - │││ -4.00 │ │ - │└┴───────┘ │ - └─────────────────┘ - ┌────────────────────────┐ - │[0]: a shape 0:1 grad_a│ - │┌┬─────────┐ │ - │││axis 0 │ │ - │├┼─────────┤ │ - │││ 1.38e+2 │ │ - │└┴─────────┘ │ - └────────────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:true `Default b; - [%expect - {| - ┌─────────────────┐ - │[2]: b shape 0:1 │ - │┌┬──────┐ │ - │││axis 0│ │ - │├┼──────┤ │ - │││ 2.00 │ │ - │└┴──────┘ │ - └─────────────────┘ - ┌────────────────────────┐ - │[2]: b shape 0:1 grad_b│ - │┌┬─────────┐ │ - │││axis 0 │ │ - │├┼─────────┤ │ - │││ 6.45e+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 + "Tnode.update_memory_mode: update 40 -> 152 inconsistent for n21") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Micrograd_demo.(fun) in file "test/micrograd_demo.ml", line 31, characters 13-67 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] let%expect_test "Micrograd half-moons example" = Tensor.unsafe_reinitialize (); @@ -180,223 +161,34 @@ let%expect_test "Micrograd half-moons example" = in Stdio.printf "Half-moons scatterplot and decision boundary:\n%!"; PrintBox_text.output Stdio.stdout plot_moons; - [%expect - {| - Half-moons scatterplot and decision boundary: - ┌────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │****************************#***********************************************************************│ - │****************************###***###***#***********************************************************│ - │***************************#********#***#***********************************************************│ - │********************#*********************##*#*#****************************************************│ - │******************#***#*****##*#*#*#****#*#*#*##****************************************************│ - │**************#*#**###**#*##**##********#**#****###**#**********************************************│ - │***********#***##***#**##*****************#*#*##*###**********************************************..│ - │***************#**###******************************##*#******************************************...│ - │*************#*****#******************************#*#***#*#*************************************....│ - │*********#***#******************......************##*###**#************************************.....│ - │******#***#*#*#****************........******************#*#*********************************.......│ - │*****###****#****************...........****************#***********************************........│ - │*******#*###****************.............******************#*#*****************************.........│ - │**********##***************.......%%......***************#**##****************************..........│ - │*****#***#****************......%..%%......*********************************************..........%.│ - │**####*******************......%..%..........****************##*#**********************........%...%│ - │****#*******************.......%%..%..........***************#*#**********************.........%....│ - │*#****##***************.........%....%.........************##***#********************.......%...%%..│ - │****#*#***************.........%..%.%...........*************#**#*******************..........%.%...│ - │##*##****************............%...............**************#*##****************.........%.%.%%%.│ - │**#####************............%...%%.............*******************************...........%..%.%..│ - │******************...............%.%.%%............*********#*#*#*#*************................%%..│ - │***##************................%..%..%............************#*#************.............%...%...│ - │##**#***********...................%...%.............********#**###***********...........%.%%.......│ - │###************..................%.%.%................***********##**********..................%....│ - │**##**********.........................................**********#*********.............%%.%..%.....│ - │**#**********......................%%...................********##********..............%%.%........│ - │************..........................%..................****#**#********...................%.......│ - │***********.........................%%.%..................*****##******...............%..%..........│ - │**********.............................%%..................***********.................%%.%%........│ - │*********..............................%...%................*******.....................%%%.........│ - │********...................................%..%....................................%.....%..........│ - │*******.................................%%..%.%................................%.%.%.%%.............│ - │******.....................................%%.%...%%.%.......................%%%....%.%.%...........│ - │*****....................................%.....%.%..%....................%.....%%.%.%...............│ - │****........................................%%...%.%%%%..%.......%.........%%.%%....................│ - │***...............................................%..%.......%..%%%%...%.%..%.%.....................│ - │**.................................................%.%.%............%..%...%........................│ - │*.................................................%....%.%......%%...%.%............................│ - │..........................................................%..%......%%%.............................│ - └────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; Stdio.printf "Loss:\n%!"; let plot_loss = PrintBox_utils.plot ~x_label:"step" ~y_label:"loss" [ Line_plot { points = Array.of_list_rev !losses; content = PrintBox.line "-" } ] in PrintBox_text.output Stdio.stdout plot_loss; - [%expect - {| - Loss: - ┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 3.79e+1│- │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │l │ │ - │o │ │ - │s │ │ - │s │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │- │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │- - - │ - │ 0.00 │----------------------------------------------------------------------------------------------------│ - ├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │0.00 3.99e+2│ - │ │ step │ - └────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; 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: - ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │ 3.63 │- │ - │ │ │ - │ │ │ - │ │ │ - │ │- │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │- │ - │ │ - - │ - │ │ - - - │ - │ │------ - │ - │ │ ---- - - --- - - - - - │ - │ │ -- - - - --- -- - -- - │ - │ │ - -- - - --- - - --- ----- - -- │ - │l │ -- - - - - --- - - - │ - │o │ - - - -- - -- - - - - - │ - │g │ - - - - -- - - - - - - - │ - │ │ - - - -- - - - -- - - - │ - │l │ - - -- -- - - - - -│ - │o │ - - - - - - - - - - - │ - │s │ - - - - -- - │ - │s │ - - │ - │ │ - - - - - - - - - │ - │ │ - - - - - │ - │ │ - - │ - │ │ - - - │ - │ │ - - │ - │ │ │ - │ │ - │ - │ │ - │ - │ │ │ - │ │ │ - │ │ │ - │ │ │ - │ │ - │ - │ │ │ - │ │ │ - │ -1.00e+1│ - - - - ----- -- --------------- ---------------------------------------------------------│ - ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │0.00 3.99e+2│ - │ │ step │ - └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; 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│--- │ - ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ - │ │0.00 3.99e+2│ - │ │ step │ - └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; (* Testing how the syntax extension %op creates labels for the resulting tensors: *) Stdio.printf "mlp_result's name: %s\n%!" @@ Tensor.debug_name mlp_result; (* Note: mlp_result is not included in the resulting tensor's label, because the identifier label does not propagate across function calls. *) - [%expect {| mlp_result's name: mlp_point |}]; + [%expect.unreachable]; (Stdio.printf "(mlp moons_input) name: %s\n%!" @@ Tensor.debug_name @@ @@ -410,4 +202,31 @@ let%expect_test "Micrograd half-moons example" = ] -> subtensor | _ -> assert false); - [%expect {| (mlp moons_input) name: mlp_moons_input |}] + [%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 40 -> 152 inconsistent for n39") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 350, characters 26-82 + Called from Tutorials__Micrograd_demo.(fun) in file "test/micrograd_demo.ml", line 137, characters 4-83 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] 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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/hello_world_op.ml.corrected diff --git a/_build/default/test/hello_world_op.ml b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/hello_world_op.ml.corrected index 9209a3a..01fe8ce 100644 --- a/_build/default/test/hello_world_op.ml +++ b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/hello_world_op.ml.corrected @@ -32,17 +32,36 @@ let%expect_test "Pointwise multiplication dims 1" = Train.forward_and_forget backend ctx y; Tensor.print ~with_code:false ~with_grad:false `Default @@ y; - [%expect - {| - ┌────────────────────┐ - │[3]: *._y shape 0:1 │ - │┌┬─────────┐ │ - │││axis 0 │ │ - │├┼─────────┤ │ - │││ 1.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. *) + (Invalid_argument + "Tnode.update_memory_mode: update 40 -> 152 inconsistent for n2") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 + Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 32, 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 "Matrix multiplication dims 1x1" = Tensor.unsafe_reinitialize (); @@ -63,29 +82,38 @@ let%expect_test "Matrix multiplication dims 1x1" = Train.forward_and_forget backend ctx y; (* Punning for ["hey"] above introduced the [hey] identifier. *) Tensor.print ~with_code:false ~with_grad:false `Default @@ hey; - [%expect - {| - ┌────────────────────────┐ - │[0]: hey shape 1:1->0:1 │ - │┌──────┬──────┐ │ - ││ │axis 1│ │ - │├──────┼──────┤ │ - ││axis 0│ 7.00 │ │ - │└──────┴──────┘ │ - └────────────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:false `Default @@ y; - [%expect - {| - ┌───────────────────┐ - │[6]: +_y shape 0:1 │ - │┌┬─────────┐ │ - │││axis 0 │ │ - │├┼─────────┤ │ - │││ 1.50e+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 40 -> 152 inconsistent for n2") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 + Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 63, 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 "Print constant tensor" = Tensor.unsafe_reinitialize (); @@ -105,49 +133,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)); @@ -158,31 +152,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 ] ]; @@ -193,29 +165,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 ] |]; @@ -226,29 +178,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 ] ] |]; @@ -259,54 +191,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 = [| [| @@ -321,63 +208,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 = [| [ @@ -392,63 +225,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 604-608, 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 107, 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 (); @@ -468,31 +260,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 1204-1206, characters 12-40 + Called from Ir__Low_level.unroll_dims.generate_all_combinations in file "arrayjit/lib/low_level.ml", lines 1204-1206, 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-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 + Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 469, 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 (); @@ -514,72 +313,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 1204-1206, characters 12-40 + Called from Ir__Low_level.unroll_dims.generate_all_combinations in file "arrayjit/lib/low_level.ml", lines 1204-1206, 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-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 + Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 515, 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 (); @@ -601,156 +369,36 @@ let%expect_test "Very big tensor" = let%op hoo = (hey * (1 + 1)) - 10 in 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 ││ - │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴──────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│ - └───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:false `Default hoo; (* Disable line wrapping for viewing the output. In VSCode: `View: Toggle Word Wrap`. *) - [%expect - {| - ┌───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │[6]: -_hoo shape 0:6|1:10,2:11 │ - │┌──────┬─────────────────────────────────────────┬─────────────────────────────────────────┬──────┬─────────────────────────────────────────┬─────────────────────────────────────────┐│ - ││ │0 @ 0 │1 @ 0 │~~~~~ │4 @ 0 │5 @ 0 ││ - ││ │axis 2 │axis 2 │axis 2│axis 2 │axis 2 ││ - │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼──────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│ - ││axis 1│ 2.53e+5 7.61e+5 ... 4.82e+6 5.33e+6 │ 5.61e+7 5.66e+7 ... 6.07e+7 6.12e+7 │ ... │ 2.23e+8 2.24e+8 ... 2.28e+8 2.28e+8 │ 2.79e+8 2.80e+8 ... 2.84e+8 2.84e+8 ││ - ││ │ 5.84e+6 6.34e+6 ... 1.04e+7 1.09e+7 │ 6.17e+7 6.22e+7 ... 6.62e+7 6.68e+7 │ │ 2.29e+8 2.29e+8 ... 2.33e+8 2.34e+8 │ 2.85e+8 2.85e+8 ... 2.89e+8 2.90e+8 ││ - ││ │ ... ... ... ... ... │ ... ... ... ... ... │ │ ... ... ... ... ... │ ... ... ... ... ... ││ - ││ │ 4.49e+7 4.54e+7 ... 4.95e+7 5.00e+7 │ 1.00e+8 1.01e+8 ... 1.05e+8 1.05e+8 │ │ 2.68e+8 2.69e+8 ... 2.73e+8 2.73e+8 │ 3.24e+8 3.24e+8 ... 3.28e+8 3.29e+8 ││ - ││ │ 5.05e+7 5.10e+7 ... 5.51e+7 5.56e+7 │ 1.06e+8 1.06e+8 ... 1.11e+8 1.11e+8 │ │ 2.74e+8 2.74e+8 ... 2.78e+8 2.79e+8 │ 3.29e+8 3.30e+8 ... 3.34e+8 3.35e+8 ││ - │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴──────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│ - └───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}] + [%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 40 -> 152 inconsistent for n1") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 622, characters 10-72 + Called from Base__List.filter_map in file "src/list.ml", line 1133, characters 11-15 + Called from Base__List.filter_map in file "src/list.ml", line 1135, characters 22-38 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml" (inlined), line 597, characters 19-42 + Called from Ir__Low_level.cleanup_virtual_llc.loop_proc in file "arrayjit/lib/low_level.ml", line 597, characters 19-64 + Called from Ir__Low_level.cleanup_virtual_llc in file "arrayjit/lib/low_level.ml", line 674, characters 36-89 + Called from Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", line 893, characters 20-94 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 508, characters 12-75 + Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 602, characters 2-42 + Called from Ppx_expect_runtime__Test_block.Configured.dump_backtrace in file "runtime/test_block.ml", line 142, characters 10-28 + |}] 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/00e0a0abf4a24803ccc3a71f5fc462de/default/test/einsum_trivia.ml.corrected diff --git a/_build/default/test/einsum_trivia.ml b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/einsum_trivia.ml.corrected index 35a36fa..bd6b49b 100644 --- a/_build/default/test/einsum_trivia.ml +++ b/_build/.sandbox/00e0a0abf4a24803ccc3a71f5fc462de/default/test/einsum_trivia.ml.corrected @@ -28,20 +28,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 {| @@ -64,106 +51,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 {| @@ -292,20 +180,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 {| @@ -363,231 +238,50 @@ let%expect_test "einsum outer product" = let%op c = (a + 1) *+ "i; j => i->j" b in 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 │ │ - │└┴────────────┘ │ - └──────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:false `Default @@ b; - [%expect - {| - ┌─────────────────────┐ - │[1]: r3 shape 0:3 │ - │┌┬──────────────────┐│ - │││axis 0 ││ - │├┼──────────────────┤│ - │││ 0.00 1.00 2.00 ││ - │└┴──────────────────┘│ - └─────────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:false `Default @@ c; - [%expect - {| - ┌──────────────────────────┐ - │[4]: ;=>_c shape 1:2->0:3 │ - │┌──────┬────────────┐ │ - ││ │axis 1 │ │ - │├──────┼────────────┤ │ - ││axis 0│ 0.00 0.00 │ │ - ││ │ 1.00 2.00 │ │ - ││ │ 2.00 4.00 │ │ - │└──────┴────────────┘ │ - └──────────────────────────┘ - |}]; + [%expect.unreachable]; let a = TDSL.range_of_shape ~batch_dims:[ 2 ] ~input_dims:[ 3 ] ~output_dims:[ 4 ] () in let b = TDSL.range_of_shape ~batch_dims:[ 5 ] ~input_dims:[ 6 ] ~output_dims:[ 7 ] () in let%op c = a *+ "i|j->k; l|m->n => il|jm->kn" b in 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 ││ - │└──────┴────────────────────────┴───────────────────────────┘│ - └─────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; 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 ││ - │└──────┴─────────────────────────────────────────┴─────────────────────────────────────────┴─────────────────────────────────────────┴─────────────────────────────────────────┴─────────────────────────────────────────┘│ - └──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:false `Default @@ c; - [%expect - {| - ┌──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐ - │[7]: ;=>_c shape 0:2,1:5|4:3,5:6->2:4,3:7 │ - │┌──────┬─────────────────────────────────────────┬─────────────────────────────────────────┬─────────────────────────────────────────┐│ - ││0 @ 1 │0 @ 4 │1 @ 4 │2 @ 4 ││ - ││ │axis 5 │axis 5 │axis 5 ││ - │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│ - ││0 @ 2 │ 0.00 0.00 ... 0.00 0.00 │ 0.00 1.00 ... 4.00 5.00 │ 0.00 2.00 ... 8.00 1.00e+1 ││ - ││axis 3│ 0.00 0.00 ... 0.00 0.00 │ 6.00 7.00 ... 1.00e+1 1.10e+1 │ 1.20e+1 1.40e+1 ... 2.00e+1 2.20e+1 ││ - ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││ - ││ │ 0.00 0.00 ... 0.00 0.00 │ 3.00e+1 3.10e+1 ... 3.40e+1 3.50e+1 │ 6.00e+1 6.20e+1 ... 6.80e+1 7.00e+1 ││ - ││ │ 0.00 0.00 ... 0.00 0.00 │ 3.60e+1 3.70e+1 ... 4.00e+1 4.10e+1 │ 7.20e+1 7.40e+1 ... 8.00e+1 8.20e+1 ││ - │├──────┼─────────────────────────────────────────┼─────────────────────────────────────────┼─────────────────────────────────────────┤│ - ││1 @ 2 │ 0.00 3.00 ... 1.20e+1 1.50e+1 │ 0.00 4.00 ... 1.60e+1 2.00e+1 │ 0.00 5.00 ... 2.00e+1 2.50e+1 ││ - ││axis 3│ 1.80e+1 2.10e+1 ... 3.00e+1 3.30e+1 │ 2.40e+1 2.80e+1 ... 4.00e+1 4.40e+1 │ 3.00e+1 3.50e+1 ... 5.00e+1 5.50e+1 ││ - ││ │ ... ... ... ... ... │ ... ... ... ... ... │ ... ... ... ... ... ││ - ││ │ 9.00e+1 9.30e+1 ... 1.02e+2 1.05e+2 │ 1.20e+2 1.24e+2 ... 1.36e+2 1.40e+2 │ 1.50e+2 1.55e+2 ... 1.70e+2 1.75e+2 ││ - ││ │ 1.08e+2 1.11e+2 ... 1.20e+2 1.23e+2 │ 1.44e+2 1.48e+2 ... 1.60e+2 1.64e+2 │ 1.80e+2 1.85e+2 ... 2.00e+2 2.05e+2 ││ ...TRUNCATED BY DUNE... - └─────────────────────────────────────────────────────────────┘ - |}]; + [%expect.unreachable]; Tensor.print ~with_code:false ~with_grad:false `Default @@ ho2; - [%expect - {| - ┌────────────────────────────────────────────────────────────────┐ - │[2]: =>_ho2 shape 0:2|2:3->1:1 │ - │┌──────┬───────────────────────────┬───────────────────────────┐│ - ││ │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 -> 36 inconsistent for r2x4x3") + Raised at Stdlib.invalid_arg in file "stdlib.ml", line 30, characters 20-45 + 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 888-889, characters 2-7 + Re-raised at Ir__Low_level.optimize_proc in file "arrayjit/lib/low_level.ml", lines 882-899, characters 30-49 + Called from Ir__Low_level.optimize in file "arrayjit/lib/low_level.ml", line 1162, characters 15-57 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 419, characters 2-85 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 412-419, characters 22-85 + 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 378, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 377-378, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 376-384, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 500, characters 25-95 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 491-504, characters 32-17 + Called from Tutorials__Einsum_trivia.(fun) in file "test/einsum_trivia.ml", line 1284, 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 (); @@ -1379,34 +612,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 {| @@ -1450,175 +659,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 │ 5.18e-20 9.16e-41 5.18e-20 ││ + ││axis 2│ 1.88e-40 1.51e-19 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 1.35e-19 4.00 ││ + ││ │ 1.51e-19 3.96e-11 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││1 @ 1 │ 5.18e-20 1.00 5.18e-20 ││ + ││axis 2│ 4.58e-41 1.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 1.00 4.00 ││ + ││ │ 1.51e-19 1.00 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││2 @ 1 │ 5.18e-20 2.00 5.18e-20 ││ + ││axis 2│ 4.58e-41 2.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 2.00 4.00 ││ + ││ │ 1.51e-19 2.00 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││3 @ 1 │ 5.18e-20 3.00 5.18e-20 ││ + ││axis 2│ 4.58e-41 3.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 3.00 4.00 ││ + ││ │ 1.51e-19 3.00 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││4 @ 1 │ 5.18e-20 4.00 5.18e-20 ││ + ││axis 2│ 1.84e-30 4.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 4.00 4.00 ││ + ││ │ 1.51e-19 4.00 5.00 ││ + │└──────┴──────────────────────────────┘│ + ├───────────────────────────────────────┤ + │┌──────┬──────────────────────────────┐│ + ││1 @ 0 │axis 3 ││ + │├──────┼──────────────────────────────┤│ + ││0 @ 1 │ 1.00 -7.38e+20 2.33e-9 ││ + ││axis 2│ 1.00 -7.38e+20 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 -7.38e+20 4.00 ││ + ││ │ 1.00 -7.38e+20 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││1 @ 1 │ 1.00 -7.38e+20 2.33e-9 ││ + ││axis 2│ 1.00 -7.38e+20 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 -7.38e+20 4.00 ││ + ││ │ 1.00 -7.38e+20 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││2 @ 1 │ 1.00 -7.38e+20 2.33e-9 ││ + ││axis 2│ 1.00 -7.38e+20 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 -7.38e+20 4.00 ││ + ││ │ 1.00 -7.38e+20 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││3 @ 1 │ 1.00 -7.38e+20 2.33e-9 ││ + ││axis 2│ 1.00 -7.38e+20 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 -7.38e+20 4.00 ││ + ││ │ 1.00 -7.38e+20 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││4 @ 1 │ 1.00 -7.38e+20 2.33e-9 ││ + ││axis 2│ 1.00 -7.38e+20 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 -7.38e+20 4.00 ││ + ││ │ 1.00 -7.38e+20 5.00 ││ + │└──────┴──────────────────────────────┘│ + ├───────────────────────────────────────┤ + │┌──────┬─────────────────────────────┐ │ + ││2 @ 0 │axis 3 │ │ + │├──────┼─────────────────────────────┤ │ + ││0 @ 1 │ 2.00 1.35e-19 1.71e-7 │ │ + ││axis 2│ 2.00 2.87e-19 1.00 │ │ + ││ │ ... ... ... │ │ + ││ │ 5.31e+22 2.71e-19 4.00 │ │ + ││ │ 2.00 3.96e-11 5.00 │ │ + │├──────┼─────────────────────────────┤ │ + ││1 @ 1 │ 2.00 1.00 1.71e-7 │ │ + ││axis 2│ 2.00 1.00 1.00 │ │ + ││ │ ... ... ... │ │ + ││ │ 5.31e+22 1.00 4.00 │ │ + ││ │ 2.00 1.00 5.00 │ │ + │├──────┼─────────────────────────────┤ │ + ││2 @ 1 │ 2.00 2.00 1.71e-7 │ │ + ││axis 2│ 2.00 2.00 1.00 │ │ + ││ │ ... ... ... │ │ + ││ │ 5.31e+22 2.00 4.00 │ │ + ││ │ 2.00 2.00 5.00 │ │ + │├──────┼─────────────────────────────┤ │ + ││3 @ 1 │ 2.00 3.00 1.71e-7 │ │ + ││axis 2│ 2.00 3.00 1.00 │ │ + ││ │ ... ... ... │ │ + ││ │ 5.31e+22 3.00 4.00 │ │ + ││ │ 2.00 3.00 5.00 │ │ + │├──────┼─────────────────────────────┤ │ + ││4 @ 1 │ 2.00 4.00 1.71e-7 │ │ + ││axis 2│ 2.00 4.00 1.00 │ │ + ││ │ ... ... ... │ │ + ││ │ 5.31e+22 4.00 4.00 │ │ + ││ │ 2.00 4.00 5.00 │ │ + │└──────┴─────────────────────────────┘ │ + ├───────────────────────────────────────┤ + │┌──────┬──────────────────────────────┐│ + ││3 @ 0 │axis 3 ││ + │├──────┼──────────────────────────────┤│ + ││0 @ 1 │ 3.00 3.96e-11 4.02e-11 ││ + ││axis 2│ 3.00 3.96e-11 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 3.96e-11 4.00 ││ + ││ │ 3.00 7.92e-11 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││1 @ 1 │ 3.00 1.00 4.02e-11 ││ + ││axis 2│ 3.00 1.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 1.00 4.00 ││ + ││ │ 3.00 1.00 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││2 @ 1 │ 3.00 2.00 4.02e-11 ││ + ││axis 2│ 3.00 2.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 2.00 4.00 ││ + ││ │ 3.00 2.00 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││3 @ 1 │ 3.00 3.00 4.02e-11 ││ + ││axis 2│ 3.00 3.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 3.00 4.00 ││ + ││ │ 3.00 3.00 5.00 ││ + │├──────┼──────────────────────────────┤│ + ││4 @ 1 │ 3.00 4.00 4.02e-11 ││ + ││axis 2│ 3.00 4.00 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 4.00 4.00 ││ + ││ │ 3.00 4.00 5.00 ││ + │└──────┴──────────────────────────────┘│ + └───────────────────────────────────────┘ |}]; 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.58e-41 5.18e-20 ││ + ││ │ 1.00 -7.38e+20 2.33e-9 ││ + ││ │ 2.00 1.35e-19 1.71e-7 ││ + ││ │ 3.00 3.96e-11 4.02e-11 ││ + │└──────┴───────────────────────────┘│ + └────────────────────────────────────┘ |}]; 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.18e-20 4.58e-41 0.00 ││ + ││ │ 4.58e-41 1.51e-19 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 5.31e+22 1.35e-19 4.00 ││ + ││ │ 1.51e-19 3.96e-11 5.00 ││ + │└──────┴──────────────────────────┘│ + └───────────────────────────────────┘ |}] let%expect_test "einsum with a leftmost input axis preserved as output axis" = @@ -1644,54 +853,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 {| @@ -1728,34 +893,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-28 12:25.52: Job failed: Failed: Build failed