2025-07-02 20:06.39: New job: test ahrefs/ocannl https://github.com/ahrefs/ocannl.git#refs/heads/master (f6776d1687e3e632ea9e715906e1221743c0c540) (linux-x86_64:alpine-3.21-5.3_opam-2.3) Base: ocaml/opam:alpine-3.21-ocaml-5.3@sha256:3343dcc8eef8cc135210194424c75ebe26927358b1a48d0c3c0860cfe43df613 Opam project build To reproduce locally: git clone --recursive "https://github.com/ahrefs/ocannl.git" -b "master" && cd "ocannl" && git reset --hard f6776d16 cat > Dockerfile <<'END-OF-DOCKERFILE' FROM ocaml/opam:alpine-3.21-ocaml-5.3@sha256:3343dcc8eef8cc135210194424c75ebe26927358b1a48d0c3c0860cfe43df613 # alpine-3.21-5.3_opam-2.3 USER 1000:1000 ENV CLICOLOR_FORCE="1" ENV OPAMCOLOR="always" WORKDIR /src 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-07-02 20:06.39: Using cache hint "ahrefs/ocannl-ocaml/opam:alpine-3.21-ocaml-5.3@sha256:3343dcc8eef8cc135210194424c75ebe26927358b1a48d0c3c0860cfe43df613-alpine-3.21-5.3_opam-2.3-82912e70cd030e6e6af979ac4425a6a2" 2025-07-02 20:06.39: Using OBuilder spec: ((from ocaml/opam:alpine-3.21-ocaml-5.3@sha256:3343dcc8eef8cc135210194424c75ebe26927358b1a48d0c3c0860cfe43df613) (comment alpine-3.21-5.3_opam-2.3) (user (uid 1000) (gid 1000)) (env CLICOLOR_FORCE 1) (env OPAMCOLOR always) (workdir /src) (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-07-02 20:06.39: Waiting for resource in pool OCluster 2025-07-02 20:06.39: Waiting for workerβ¦ 2025-07-02 20:06.40: Got resource from pool OCluster Building on asteria.caelum.ci.dev All commits already cached HEAD is now at f6776d16 Cleanup: remove unused ~v input to fetch_op (from ocaml/opam:alpine-3.21-ocaml-5.3@sha256:3343dcc8eef8cc135210194424c75ebe26927358b1a48d0c3c0860cfe43df613) 2025-07-02 20:06.41 ---> using "3141b9cf5ba038771635d2742247938bc41e269100016f85aeafaf7cfcddc947" from cache /: (comment alpine-3.21-5.3_opam-2.3) /: (user (uid 1000) (gid 1000)) /: (env CLICOLOR_FORCE 1) /: (env OPAMCOLOR always) /: (workdir /src) /src: (run (shell "sudo ln -f /usr/bin/opam-2.3 /usr/bin/opam")) 2025-07-02 20:06.41 ---> using "80b751ed47c43451e88cd0293ed9a635ccea08011eeda4114e7ec036c06c48a3" 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-07-02 20:06.41 ---> using "6c30ca4e379a499434293ac7ae4ccb6757139c93d8bc5916b7557572123e5ba0" from cache /src: (run (shell "uname -rs && opam exec -- ocaml -version && opam --version")) Linux 5.15.0-142-generic The OCaml toplevel, version 5.3.0 2.3.0 2025-07-02 20:06.41 ---> using "a3df42546ff5706a2753d5e221aadf5e7ea87ec715342ceb488042e5402600c3" from cache /src: (workdir /src) /src: (run (shell "sudo chown opam /src")) 2025-07-02 20:06.41 ---> using "17ab95ba4820eaae0552c498f14e3328c5320ed844b460f0519c8783adab646a" 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..84d2c8f5ca 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-07-02 20:06.41 ---> using "1ba3f53347b37a674b955528514f7f6e01a1336e9b769320cf0f5a2096b708ae" from cache /src: (copy (src neural_nets_lib.opam arrayjit.opam) (dst ./)) 2025-07-02 20:06.41 ---> using "c376d988bfc6d6405b3dd6dc9356e8929aab9a39ba512937d1d94455a43cc26e" 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-07-02 20:06.41 ---> using "29f29b75245e46c4bc1983d96cda1ab0a297f0c998aafbca7376690bc483f993" from cache /src: (run (network host) (shell "echo '(lang dune 3.0)' > './dune-project'")) 2025-07-02 20:06.41 ---> using "51adc441b434f811d59cb88b89a9ed890837617b8e5fb90ca5826008083ca147" 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/bin/sudo "apk" "update" - fetch https://dl-cdn.alpinelinux.org/alpine/v3.21/main/x86_64/APKINDEX.tar.gz - fetch https://dl-cdn.alpinelinux.org/alpine/v3.21/community/x86_64/APKINDEX.tar.gz - fetch https://dl-cdn.alpinelinux.org/alpine/edge/main/x86_64/APKINDEX.tar.gz - fetch https://dl-cdn.alpinelinux.org/alpine/edge/community/x86_64/APKINDEX.tar.gz - fetch https://dl-cdn.alpinelinux.org/alpine/edge/testing/x86_64/APKINDEX.tar.gz - v3.21.3-612-gd8580ac27b8 [https://dl-cdn.alpinelinux.org/alpine/v3.21/main] - v3.21.3-615-g437846cb9e3 [https://dl-cdn.alpinelinux.org/alpine/v3.21/community] - v3.23.0_alpha20250612-1251-ga7e5898a0de [https://dl-cdn.alpinelinux.org/alpine/edge/main] - v3.23.0_alpha20250612-1255-g184493ceb19 [https://dl-cdn.alpinelinux.org/alpine/edge/community] - v3.23.0_alpha20250612-1226-ga3dc5664999 [https://dl-cdn.alpinelinux.org/alpine/edge/testing] - OK: 58864 distinct packages available <><> Synchronising pinned packages ><><><><><><><><><><><><><><><><><><><><><><> [arrayjit.dev] synchronised (file:///src) [neural_nets_lib.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: curl-dev libffi-dev zlib-dev <><> Handling external dependencies <><><><><><><><><><><><><><><><><><><><><><> + /usr/bin/sudo "apk" "add" "curl-dev" "libffi-dev" "zlib-dev" - (1/13) Installing brotli (1.1.0-r2) - (2/13) Installing brotli-dev (1.1.0-r2) - (3/13) Installing c-ares-dev (1.34.5-r0) - (4/13) Installing libidn2-dev (2.3.7-r0) - (5/13) Installing libpsl-utils (0.21.5-r3) - (6/13) Installing libpsl-dev (0.21.5-r3) - (7/13) Installing nghttp2-dev (1.64.0-r0) - (8/13) Installing openssl-dev (3.3.3-r0) - (9/13) Installing zlib-dev (1.3.1-r2) - (10/13) Installing zstd-dev (1.5.6-r2) - (11/13) Installing curl-dev (8.12.1-r1) - (12/13) Installing linux-headers (6.6-r1) - (13/13) Installing libffi-dev (3.4.7-r0) - Executing busybox-1.37.0-r12.trigger - OK: 317 MiB in 115 packages 2025-07-02 20:06.41 ---> using "5258eca8459b3b9d6b3d72f50acbf60291e55a426dbaa2a8a05b24f63514a0d4" 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 bigstringaf.0.10.0 (cached) -> retrieved base.v0.17.3 (cached) -> retrieved camlp-streams.5.0.1 (cached) -> retrieved camlzip.1.13 (cached) -> installed conf-libcurl.2 -> retrieved cmdliner.1.3.0 (cached) -> retrieved cppo.1.8.0 (cached) -> 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 multicore-magic.2.3.1 (cached) -> retrieved mdx.2.5.0 (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 ocamlfind.1.9.8 (cached) -> retrieved ocamlbuild.0.16.1 (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 dune.3.19.1, dune-configurator.3.19.1 (cached) -> retrieved ppx_compare.v0.17.0 (cached) -> retrieved ppx_derivers.1.2.1 (cached) -> retrieved ppx_deriving.6.1.0 (cached) -> retrieved ppx_enumerate.v0.17.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) -> installed cmdliner.1.3.0 -> installed num.1.6 -> retrieved ppx_here.v0.17.0 (cached) -> retrieved ppx_inline_test.v0.17.1 (cached) -> retrieved ppx_optcomp.v0.17.1 (cached) -> retrieved ppx_minidebug.2.3.0 (cached) -> retrieved ppx_sexp_conv.v0.17.1 (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) -> retrieved ppxlib.0.36.0 (cached) -> retrieved result.1.5 (cached) -> retrieved seq.base (cached) -> installed seq.base -> retrieved sexplib.v0.17.0 (cached) -> retrieved saturn_lockfree.0.5.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 ocamlfind.1.9.8 -> installed base-bytes.base -> installed camlzip.1.13 -> installed ocamlbuild.0.16.1 -> installed topkg.1.0.8 -> installed uutf.1.0.4 -> installed mtime.2.1.0 -> 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 cppo.1.8.0 -> installed csv.2.4 -> installed multicore-magic.2.3.1 -> installed ocaml-compiler-libs.v0.17.0 -> installed ocaml-syntax-shims.1.0.0 -> installed ocaml-version.4.0.1 -> installed ocaml_intrinsics_kernel.v0.17.1 -> installed pprint.20230830 -> installed printbox.0.12 -> installed re.1.12.0 -> installed result.1.5 -> installed sexplib0.v0.17.0 -> installed stdlib-shims.0.3.0 -> installed thread-local-storage.0.2 -> installed saturn_lockfree.0.5.0 -> installed integers.0.7.0 -> installed parsexp.v0.17.0 -> installed dune-configurator.3.19.1 -> installed bigstringaf.0.10.0 -> installed mdx.2.5.0 -> installed sexplib.v0.17.0 -> installed tyxml.4.6.0 -> installed angstrom.0.16.1 -> installed printbox-html.0.12 -> installed curl.0.10.0 -> installed ctypes.0.23.0 -> 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 uucp.16.0.0 -> installed printbox-text.0.12 -> installed printbox-md.0.12 -> installed printbox-ext-plot.0.12 -> installed ppxlib.0.36.0 -> installed ppx_optcomp.v0.17.1 -> installed ppxlib_jane.v0.17.4 -> installed ppx_cold.v0.17.0 -> installed ppx_here.v0.17.0 -> installed ppx_variants_conv.v0.17.1 -> installed ppx_fields_conv.v0.17.0 -> installed ppx_enumerate.v0.17.0 -> installed ppx_globalize.v0.17.2 -> installed ppx_deriving.6.1.0 -> 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_minidebug.2.3.0 -> installed ppx_base.v0.17.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-07-02 20:06.41 ---> using "d46f3b58419fd0d277bd1e1f5446a5f7209bb2c9d5422ed4ba8702570dbbb56e" from cache /src: (copy (src .) (dst /src)) 2025-07-02 20:06.41 ---> saved as "18c55dcc62a0a1777bfc694cfe118e47fd5c9ea78f9883ec0f9e2b85b26bcab7" /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 File "test/einsum/test_conv_syntax.expected", line 1, characters 0-0: /usr/bin/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... File "test/test_print_style.expected", line 1, characters 0-0: /usr/bin/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/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 (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/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 #35} 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 128)))) @!Fatal error: exception (lib/shape.ml.Shape_error "apply_row_constraint: Total_elems constraint failed, shape is too small" ((Row_mismatch ((dims ((Dim ((d 1200) (label ()) (proj_id ()))) (Dim ((d 2) (label ()) (proj_id ()))))) (bcast Broadcastable) (id ((sh_id 13) (kind Batch))))))) Raised at Ocannl__Row.apply_row_constraint in file "lib/row.ml", lines 487-490, characters 16-47 Called from Ocannl__Row.solve_inequalities.solve.f in file "lib/row.ml", line 1494, characters 32-72 Called from Base__List0.fold in file "src/list0.ml", line 43, characters 27-37 Called from Ocannl__Row.solve_inequalities.solve in file "lib/row.ml", line 1503, characters 22-56 Re-raised at Ocannl__Row.solve_inequalities.solve in file "lib/row.ml", lines 1456-1509, characters 16-25 Called from Ocannl__Row.solve_inequalities in file "lib/row.ml", line 1512, characters 50-65 Re-raised at Ocannl__Row.solve_inequalities in file "lib/row.ml", lines 1454-1535, characters 35-44 Called from Ocannl__Shape.propagate_shapes in file "lib/shape.ml", line 657, characters 20-73 Re-raised at Ocannl__Shape.propagate_shapes in file "lib/shape.ml", lines 650-660, characters 33-14 Called from Base__List0.iter in file "src/list0.ml", line 66, characters 4-7 Called from Ocannl__Tensor.op in file "lib/tensor.ml", line 235, characters 2-57 Called from Ocannl__Train.example_train_loop in file "lib/train.ml", line 387, characters 15-58 Called from Dune__exe__Moons_demo_parallel_run.main in file "test/moons_demo_parallel_run.ml", lines 65-69, characters 4-8 Called from Base__Exn.protectx in file "src/exn.ml", line 79, characters 8-11 Re-raised at Base__Exn.raise_with_original_backtrace in file "src/exn.ml" (inlined), line 59, characters 2-50 Called from Base__Exn.protectx in file "src/exn.ml", line 86, characters 13-49 Called from Base__Exn.protect in file "src/exn.ml" (inlined), line 92, characters 26-49 Called from Dune__exe__Moons_demo_parallel_run in file "test/moons_demo_parallel_run.ml", line 90, characters 2-53 (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition 'Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/62fab142bc0e3ba6e97991a28e495d99/default/test/ocannl_config.' -source-tree-root .. -diff-cmd -) Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/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_limit_constant_fill_size Not found, using default 256 Retrieving commandline, environment, or config file variable ocannl_cd_ident_style Not found, using default heuristic Retrieving commandline, environment, or config file variable ocannl_ll_ident_style Not found, using default heuristic Retrieving commandline, environment, or config file variable ocannl_prefer_backend_uniformity Found true, in the config file Retrieving commandline, environment, or config file variable ocannl_debug_log_to_stream_files Found true, commandline --ocannl_debug_log_to_stream_files=true Retrieving commandline, environment, or config file variable ocannl_cc_backend_optimization_level Not found, using default 3 Retrieving commandline, environment, or config file variable ocannl_cc_backend_compiler_command Not found, using default gcc Retrieving commandline, environment, or config file variable ocannl_cc_backend_verify_codesign Not found, using default false Fatal error: exception File "src/printbox-text/PrintBox_text.ml", line 212, characters 6-12: Assertion failed Raised at PrintBox_text.Output.Make_out.to_buf_aux_ in file "src/printbox-text/PrintBox_text.ml", line 212, characters 6-50 Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 19-42 Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41 Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41 Called from Stdlib__Map.Make.fold in file "map.ml", line 329, characters 26-41 Called from PrintBox_text.Output.Make_out.render in file "src/printbox-text/PrintBox_text.ml", line 242, characters 14-64 Called from PrintBox_text.output in file "src/printbox-text/PrintBox_text.ml", line 851, characters 2-31 Called from Minidebug_runtime.PrintBox.output_box in file "minidebug_runtime.ml", line 1527, characters 19-59 Called from Minidebug_runtime.PrintBox.close_log_impl.close_tree in file "minidebug_runtime.ml", line 1572, characters 6-38 Called from Backends.Add_buffer_retrieval_and_syncing.sync_routine in file "arrayjit/lib/backends.ml", lines 207-235, characters 31-82 Called from Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 539-540, characters 4-92 Re-raised at Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 526-540, characters 23-92 Called from Dune__exe__Micrograd_demo_logging in file "test/micrograd_demo_logging.ml", line 36, characters 13-76 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 Utils.User_error("Constant_fill size is too large to unroll for moons_classes (size: 400, limit: 256), either increase ocannl_limit_constant_fill_size or use Tnode.set_values instead") Raised at Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", lines 276-281, characters 10-83 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 250, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 247, characters 66-72 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 247, characters 66-72 Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 Called from Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", line 285, characters 2-11 Re-raised at Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", lines 152-285, characters 29-11 Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 448, characters 26-43 Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 448, characters 2-43 Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 446-453, characters 22-85 Called from Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 256-257, characters 4-45 Re-raised at Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 248-257, characters 34-47 Called from Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", line 441, characters 6-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 440-441, characters 4-59 Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 439-447, characters 26-99 Called from Ocannl__Train.to_routine in file "lib/train.ml", line 354, characters 26-82 Called from Dune__exe__Moons_demo_variant in file "test/einsum/moons_demo_variant.ml", line 64, characters 4-100 (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file File "test/einsum/einsum_trivia_exec.expected", line 1, characters 0-0: /usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum/einsum_trivia_exec.expected _build/default/test/einsum/einsum_trivia_exec.exe.output diff --git a/_build/default/test/einsum/einsum_trivia_exec.expected b/_build/default/test/einsum/einsum_trivia_exec.exe.output index ab456b5..74ceb2e 100644 --- a/_build/default/test/einsum/einsum_trivia_exec.expected +++ b/_build/default/test/einsum/einsum_trivia_exec.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 HERE: test/einsum/einsum_trivia_exec.ml:31:21 (cd _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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/f018ead6d62af87c68db42e044a51bc0/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/bin/git --no-pager diff --no-index --color=always -u _build/default/test/zero2hero_1of7.ml _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/zero2hero_1of7.ml.corrected diff --git a/_build/default/test/zero2hero_1of7.ml b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/zero2hero_1of7.ml.corrected index 3b7cc00..394b619 100644 --- a/_build/default/test/zero2hero_1of7.ml +++ b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/zero2hero_1of7.ml.corrected @@ -30,24 +30,24 @@ let%expect_test "Graph drawing recompile" = 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> β β β + #15 +_f_nd + 5.00 + #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] β + <void> β<void> β<void> β β + β#8 grad_**. Virt/30β β β + β<void> β β β + β[0]β #6 2. Virt/40 β β β + β β <void> β β β |}]; let%op f = (3 *. ("x" [ 5 ] **. 2)) - (4 *. x) + 5 in Train.every_non_literal_on_host f; @@ -55,27 +55,7 @@ let%expect_test "Graph drawing recompile" = 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 +74,21 @@ 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. *) + ("Utils.User_error(\"The linked context lacks node x\")") + Raised at Backends.verify_prior_context.(fun) in file "arrayjit/lib/backends.ml", line 285, characters 11-93 + Called from Base__Set.Accessors.iter in file "src/set.ml" (inlined), line 1193, characters 18-38 + Called from Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 281-285, characters 2-94 + Re-raised at Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 280-285, characters 37-94 + Called from Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 527-528, characters 4-49 + Re-raised at Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 526-540, characters 23-92 + Called from Tutorials__Zero2hero_1of7.(fun) in file "test/zero2hero_1of7.ml", line 55, characters 16-69 + 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 (); @@ -271,96 +218,39 @@ 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. *) + ("Utils.User_error(\"The linked context lacks node a\")") + Raised at Backends.verify_prior_context.(fun) in file "arrayjit/lib/backends.ml", line 285, characters 11-93 + Called from Base__Set.Tree0.iter.iter in file "src/set.ml", line 860, characters 8-14 + Called from Base__Set.Accessors.iter in file "src/set.ml" (inlined), line 1193, characters 18-38 + Called from Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 281-285, characters 2-94 + Re-raised at Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 280-285, characters 37-94 + Called from Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 527-528, characters 4-49 + Re-raised at Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 526-540, characters 23-92 + 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 (); @@ -381,68 +271,29 @@ let%expect_test "Simple gradients virtual" = Tensor.print_tree ~spy:true ~with_grad:true ~depth:9 l; [%expect {| - #12 *._l Host&dev/41 - <not-in-yet> - #13 grad_*._l unknown - <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 - <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 β - <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 β β - <not-in-yet> β<not-in-yet> β β + #12 *._l Host&dev/412 + <not-in-yet> + #13 grad_*._l unknown + <not-in-yet> + #8 +_d unknown β[10] + <not-in-yet> β + #9 grad_+_d unknown β + <not-in-yet> β + #4 *._e unknown β[6]β + <not-in-yet> β β + #5 grad_*._e unknownβ β + <not-in-yet> β β + [0]β[2] β β |}]; 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 +301,28 @@ 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. *) + ("Utils.User_error(\"The linked context lacks node a\")") + Raised at Backends.verify_prior_context.(fun) in file "arrayjit/lib/backends.ml", line 285, characters 11-93 + Called from Base__Set.Tree0.iter.iter in file "src/set.ml", line 860, characters 8-14 + Called from Base__Set.Tree0.iter.iter in file "src/set.ml", line 860, characters 8-14 + Called from Base__Set.Accessors.iter in file "src/set.ml" (inlined), line 1193, characters 18-38 + Called from Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 281-285, characters 2-94 + Re-raised at Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 280-285, characters 37-94 + Called from Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 527-528, characters 4-49 + Re-raised at Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 526-540, characters 23-92 + 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 +341,22 @@ 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. *) + ("Utils.User_error(\"The linked context lacks node b\")") + Raised at Backends.verify_prior_context.(fun) in file "arrayjit/lib/backends.ml", line 285, characters 11-93 + Called from Base__Set.Tree0.iter.iter in file "src/set.ml", line 861, characters 8-11 + Called from Base__Set.Accessors.iter in file "src/set.ml" (inlined), line 1193, characters 18-38 + Called from Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 281-285, characters 2-94 + Re-raised at Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 280-285, characters 37-94 + Called from Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 527-528, characters 4-49 + Re-raised at Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 526-540, characters 23-92 + 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 +369,19 @@ 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. *) + ("Utils.User_error(\"The linked context lacks node b\")") + Raised at Backends.verify_prior_context.(fun) in file "arrayjit/lib/backends.ml", line 285, characters 11-93 + Called from Base__Set.Tree0.iter.iter in file "src/set.ml", line 860, characters 8-14 + Called from Base__Set.Accessors.iter in file "src/set.ml" (inlined), line 1193, characters 18-38 + Called from Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 281-285, characters 2-94 + Re-raised at Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 280-285, characters 37-94 + Called from Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 527-528, characters 4-49 + Re-raised at Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 526-540, characters 23-92 + 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/moons_demo_parallel.ml", line 1, characters 0-0: /usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/moons_demo_parallel.ml _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/moons_demo_parallel.ml.corrected diff --git a/_build/default/test/moons_demo_parallel.ml b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/moons_demo_parallel.ml.corrected index ce09725..04904f9 100644 --- a/_build/default/test/moons_demo_parallel.ml +++ b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/moons_demo_parallel.ml.corrected @@ -105,4 +105,30 @@ 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. *) + (lib/shape.ml.Shape_error + "apply_row_constraint: Total_elems constraint failed, shape is too small" + ((Row_mismatch + ((dims + ((Dim ((d 1200) (label ()) (proj_id ()))) + (Dim ((d 2) (label ()) (proj_id ()))))) + (bcast Broadcastable) (id ((sh_id 12) (kind Batch))))))) + Raised at Ocannl__Row.apply_row_constraint in file "lib/row.ml", lines 487-490, characters 16-47 + Called from Ocannl__Row.solve_inequalities.solve.f in file "lib/row.ml", line 1494, characters 32-72 + Called from Base__List0.fold in file "src/list0.ml", line 43, characters 27-37 + Called from Ocannl__Row.solve_inequalities.solve in file "lib/row.ml", line 1503, characters 22-56 + Re-raised at Ocannl__Row.solve_inequalities.solve in file "lib/row.ml", lines 1456-1509, characters 16-25 + Called from Ocannl__Row.solve_inequalities in file "lib/row.ml", line 1512, characters 50-65 + Re-raised at Ocannl__Row.solve_inequalities in file "lib/row.ml", lines 1454-1535, characters 35-44 + Called from Ocannl__Shape.propagate_shapes in file "lib/shape.ml", line 657, characters 20-73 + Re-raised at Ocannl__Shape.propagate_shapes in file "lib/shape.ml", lines 650-660, characters 33-14 + Called from Base__List0.iter in file "src/list0.ml", line 66, characters 4-7 + Called from Ocannl__Tensor.op in file "lib/tensor.ml", line 235, characters 2-57 + Called from Ocannl__Train.example_train_loop in file "lib/train.ml", line 387, characters 15-58 + 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/bin/git --no-pager diff --no-index --color=always -u _build/default/test/micrograd_demo.ml _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/micrograd_demo.ml.corrected diff --git a/_build/default/test/micrograd_demo.ml b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/micrograd_demo.ml.corrected index 59a566d..92074e9 100644 --- a/_build/default/test/micrograd_demo.ml +++ b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/micrograd_demo.ml.corrected @@ -31,57 +31,26 @@ 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. *) + ("Utils.User_error(\"The linked context lacks node a\")") + Raised at Backends.verify_prior_context.(fun) in file "arrayjit/lib/backends.ml", line 285, characters 11-93 + Called from Base__Set.Tree0.iter.iter in file "src/set.ml", line 860, characters 8-14 + Called from Base__Set.Accessors.iter in file "src/set.ml" (inlined), line 1193, characters 18-38 + Called from Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 281-285, characters 2-94 + Re-raised at Backends.verify_prior_context in file "arrayjit/lib/backends.ml", lines 280-285, characters 37-94 + Called from Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 527-528, characters 4-49 + Re-raised at Backends.Raise_backend.link in file "arrayjit/lib/backends.ml", lines 526-540, characters 23-92 + 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 +149,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 +190,37 @@ 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. *) + ("Utils.User_error(\"Constant_fill size is too large to unroll for moons_classes (size: 400, limit: 256), either increase ocannl_limit_constant_fill_size or use Tnode.set_values instead\")") + Raised at Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", lines 276-281, characters 10-83 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 250, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 247, characters 66-72 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 247, characters 66-72 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 249, characters 17-24 + Called from Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", line 285, characters 2-11 + Re-raised at Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", lines 152-285, characters 29-11 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 448, characters 26-43 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 448, characters 2-43 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 446-453, characters 22-85 + Called from Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 256-257, characters 4-45 + Re-raised at Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 248-257, characters 34-47 + Called from Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", line 441, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 440-441, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 439-447, characters 26-99 + Called from Ocannl__Train.to_routine in file "lib/train.ml", line 354, 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/bin/git --no-pager diff --no-index --color=always -u _build/default/test/hello_world_op.ml _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/hello_world_op.ml.corrected diff --git a/_build/default/test/hello_world_op.ml b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/hello_world_op.ml.corrected index 5674865..6aa498b 100644 --- a/_build/default/test/hello_world_op.ml +++ b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/hello_world_op.ml.corrected @@ -36,11 +36,11 @@ let%expect_test "Pointwise multiplication dims 1" = {| ββββββββββββββββββββββ β[3]: *._y shape 0:1 β - βββ¬ββββββββββ β - βββaxis 0 β β - βββΌββββββββββ€ β - βββ 1.40e+1 β β - βββ΄ββββββββββ β + βββ¬βββββββ β + βββaxis 0β β + βββΌβββββββ€ β + βββ 0.00 β β + βββ΄βββββββ β ββββββββββββββββββββββ |}] @@ -70,7 +70,7 @@ let%expect_test "Matrix multiplication dims 1x1" = βββββββββ¬βββββββ β ββ βaxis 1β β βββββββββΌβββββββ€ β - ββaxis 0β 7.00 β β + ββaxis 0β 0.00 β β βββββββββ΄βββββββ β ββββββββββββββββββββββββββ |}]; @@ -79,11 +79,11 @@ let%expect_test "Matrix multiplication dims 1x1" = {| βββββββββββββββββββββ β[6]: +_y shape 0:1 β - βββ¬ββββββββββ β - βββaxis 0 β β - βββΌββββββββββ€ β - βββ 1.50e+1 β β - βββ΄ββββββββββ β + βββ¬βββββββ β + βββaxis 0β β + βββΌβββββββ€ β + βββ 1.00 β β + βββ΄βββββββ β βββββββββββββββββββββ |}] @@ -108,7 +108,7 @@ let%expect_test "Print constant tensor" = 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 [ + [0]: c2x3_hey shape 1:3->0:2 [ 1.00 , 2.00 , 3.00 ; 4.00 , 5.00 , 6.00 ] @@ -116,22 +116,22 @@ let%expect_test "Print constant tensor" = 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 β β - βββββββββ΄βββββββββββββββββββ β - ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ + βββββββββββββββββββββββββββββββ + β[0]: c2x3_hey shape 1:3->0:2 β + βββββββββ¬βββββββββββββββββββ β + ββ βaxis 1 β β + βββββββββΌβββββββββββββββββββ€ β + ββaxis 0β 1.00 2.00 3.00 β β + ββ β 4.00 5.00 6.00 β β + βββββββββ΄βββββββββββββββββββ β + βββββββββββββββββββββββββββββββ |}]; 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]: c2x3_hoo shape 0:2|1:3 [| [ 1.00 ; 2.00 ; 3.00 ] ; [ 4.00 ; 5.00 ; 6.00 ] |] @@ -139,15 +139,15 @@ let%expect_test "Print constant tensor" = 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 β β - βββββββββ΄βββββββββββββββββββ β - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ + ββββββββββββββββββββββββββββββ + β[1]: c2x3_hoo shape 0:2|1:3 β + βββββββββ¬βββββββββββββββββββ β + ββ βaxis 1 β β + βββββββββΌβββββββββββββββββββ€ β + ββaxis 0β 1.00 2.00 3.00 β β + ββ β 4.00 5.00 6.00 β β + βββββββββ΄βββββββββββββββββββ β + ββββββββββββββββββββββββββββββ |}]; let%op hey2 = [ @@ -469,31 +469,51 @@ 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. *) + (lib/shape.ml.Shape_error + "apply_row_constraint: Total_elems constraint failed, shape is too big" + ((Dim_mismatch))) + Raised at Ocannl__Row.apply_row_constraint in file "lib/row.ml", lines 479-482, characters 12-42 + Called from Ocannl__Row.solve_inequalities.solve.f in file "lib/row.ml", line 1494, characters 32-72 + Called from Base__List0.fold in file "src/list0.ml", line 43, characters 27-37 + Called from Ocannl__Row.solve_inequalities.solve in file "lib/row.ml", line 1503, characters 22-56 + Re-raised at Ocannl__Row.solve_inequalities.solve in file "lib/row.ml", lines 1456-1509, characters 16-25 + Called from Ocannl__Row.solve_inequalities in file "lib/row.ml", line 1512, characters 50-65 + Re-raised at Ocannl__Row.solve_inequalities in file "lib/row.ml", lines 1454-1535, characters 35-44 + Called from Ocannl__Shape.derive_projections in file "lib/shape.ml", line 785, characters 28-84 + Re-raised at Ocannl__Shape.derive_projections in file "lib/shape.ml", lines 774-841, characters 35-70 + Called from CamlinternalLazy.do_force_block in file "camlinternalLazy.ml", line 49, characters 17-27 + Re-raised at CamlinternalLazy.do_force_block in file "camlinternalLazy.ml", line 56, characters 4-11 + Called from CamlinternalLazy.force_lazy_block in file "camlinternalLazy.ml" (inlined), line 78, characters 27-67 + Called from Ocannl__Tensor.term.op_asn in file "lib/tensor.ml", line 343, characters 20-44 + Called from CamlinternalLazy.do_force_block in file "camlinternalLazy.ml", line 49, characters 17-27 + Re-raised at CamlinternalLazy.do_force_block in file "camlinternalLazy.ml", line 56, characters 4-11 + Called from CamlinternalLazy.force_lazy_block in file "camlinternalLazy.ml" (inlined), line 78, characters 27-67 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 269, characters 61-72 + Called from Ir__Assignments.to_low_level.loop in file "arrayjit/lib/assignments.ml", line 247, characters 66-72 + Called from Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", line 285, characters 2-11 + Re-raised at Ir__Assignments.to_low_level in file "arrayjit/lib/assignments.ml", lines 152-285, characters 29-11 + Called from Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 448, characters 26-43 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", line 448, characters 2-43 + Re-raised at Ir__Assignments.lower in file "arrayjit/lib/assignments.ml", lines 446-453, characters 22-85 + Called from Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 256-257, characters 4-45 + Re-raised at Backends.lower_assignments in file "arrayjit/lib/backends.ml", lines 248-257, characters 34-47 + Called from Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", line 441, characters 6-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 440-441, characters 4-59 + Re-raised at Backends.Raise_backend.compile in file "arrayjit/lib/backends.ml", lines 439-447, characters 26-99 + Called from Ocannl__Train.forward_and_ctx in file "lib/train.ml", line 520, characters 37-90 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 519-522, characters 6-18 + Re-raised at Ocannl__Train.forward_and_ctx in file "lib/train.ml", lines 507-529, characters 32-17 + Called from Ocannl__Train.forward_and_forget in file "lib/train.ml", line 533, characters 12-94 + Called from Tutorials__Hello_world_op.(fun) in file "test/hello_world_op.ml", line 470, 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 (); @@ -515,71 +535,38 @@ 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 <virtual> |}]; + Tensor.print ~with_code:false ~with_grad:false `Default zero_to_twenty; + [%expect + {| [2]: 0...20 shape 0:21 <virtual> |}]; + Tensor.print ~with_code:false ~with_grad:false `Default hey; [%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 - ] + ββββββββββββββββββββββββββββββββββββββββ + β[0]: hey shape 1:21->0:21 β + βββββββββ¬βββββββββββββββββββββββββββββββ + ββ βaxis 1 ββ + βββββββββΌββββββββββββββββββββββββββββββ€β + ββaxis 0β 0.00 0.00 ... 0.00 0.00 ββ + ββ β 0.00 0.00 ... 0.00 0.00 ββ + ββ β ... ... ... ... ... ββ + ββ β 0.00 0.00 ... 0.00 0.00 ββ + ββ β 0.00 0.00 ... 0.00 0.00 ββ + βββββββββ΄βββββββββββββββββββββββββββββββ + ββββββββββββββββββββββββββββββββββββββββ |}]; - Tensor.print ~with_code:false ~with_grad:false `Default zero_to_twenty; + Tensor.print ~with_code:false ~with_grad:false `Default y; [%expect {| ββββββββββββββββββββββββββββββββββββββββ - β[2]: 0...20 shape 0:21 β + β[5]: + shape 0:21 β βββ¬βββββββββββββββββββββββββββββββββββββ βββaxis 0 ββ βββΌββββββββββββββββββββββββββββββββββββ€β βββ 0.00 1.00 ... 1.90e+1 2.00e+1 ββ βββ΄βββββββββββββββββββββββββββββββββββββ ββββββββββββββββββββββββββββββββββββββββ - |}]; - 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 ββ - βββββββββ΄βββββββββββββββββββββββββββββββββββββββββββ - ββββββββββββββββββββββββββββββββββββββββββββββββββββ - |}]; - 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 ββ - βββ΄βββββββββββββββββββββββββββββββββββββββββββ - ββββββββββββββββββββββββββββββββββββββββββββββ |}] let%expect_test "Very big tensor" = @@ -603,140 +590,7 @@ let%expect_test "Very big tensor" = Train.forward_and_forget backend ctx hoo; Tensor.print ~with_code:false ~with_grad:false `Default hey; [%expect - {| - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ - β[0]: r6x10x11x7x8x9 shape 0:6|3:7,4:8,5:9->1:10,2:11 β - βββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββββββββββββββββββββββββββββββββββββββ - ββ0 @ 0 β0 @ 4 β1 @ 4 β~~~~~ β6 @ 4 β7 @ 4 ββ - ββ βaxis 5 βaxis 5 βaxis 5βaxis 5 βaxis 5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ0 @ 1 β 0.00 1.00 ... 7.00 8.00 β 9.00 1.00e+1 ... 1.60e+1 1.70e+1 β ... β 5.40e+1 5.50e+1 ... 6.10e+1 6.20e+1 β 6.30e+1 6.40e+1 ... 7.00e+1 7.10e+1 ββ - ββaxis 2β 5.04e+2 5.05e+2 ... 5.11e+2 5.12e+2 β 5.13e+2 5.14e+2 ... 5.20e+2 5.21e+2 β β 5.58e+2 5.59e+2 ... 5.65e+2 5.66e+2 β 5.67e+2 5.68e+2 ... 5.74e+2 5.75e+2 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 4.53e+3 4.53e+3 ... 4.54e+3 4.54e+3 β 4.54e+3 4.54e+3 ... 4.55e+3 4.55e+3 β β 4.59e+3 4.59e+3 ... 4.59e+3 4.59e+3 β 4.59e+3 4.60e+3 ... 4.60e+3 4.60e+3 ββ - ββ β 5.04e+3 5.04e+3 ... 5.04e+3 5.04e+3 β 5.04e+3 5.05e+3 ... 5.05e+3 5.05e+3 β β 5.09e+3 5.09e+3 ... 5.10e+3 5.10e+3 β 5.10e+3 5.10e+3 ... 5.11e+3 5.11e+3 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ1 @ 1 β 5.54e+3 5.54e+3 ... 5.55e+3 5.55e+3 β 5.55e+3 5.55e+3 ... 5.56e+3 5.56e+3 β ... β 5.59e+3 5.59e+3 ... 5.60e+3 5.60e+3 β 5.60e+3 5.60e+3 ... 5.61e+3 5.61e+3 ββ - ββaxis 2β 6.04e+3 6.04e+3 ... 6.05e+3 6.05e+3 β 6.05e+3 6.05e+3 ... 6.06e+3 6.06e+3 β β 6.10e+3 6.10e+3 ... 6.10e+3 6.11e+3 β 6.11e+3 6.11e+3 ... 6.11e+3 6.11e+3 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 1.00e+4 1.00e+4 ... 1.00e+4 1.00e+4 β 1.00e+4 1.00e+4 ... 1.00e+4 1.00e+4 β β 1.01e+4 1.01e+4 ... 1.01e+4 1.01e+4 β 1.01e+4 1.01e+4 ... 1.01e+4 1.01e+4 ββ - ββ β 1.05e+4 1.05e+4 ... 1.05e+4 1.05e+4 β 1.05e+4 1.05e+4 ... 1.06e+4 1.06e+4 β β 1.06e+4 1.06e+4 ... 1.06e+4 1.06e+4 β 1.06e+4 1.06e+4 ... 1.06e+4 1.06e+4 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ~~~~~ β ... β ... β ... β ... β ... ββ - ββaxis 2β β β β β ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ8 @ 1 β 4.43e+4 4.43e+4 ... 4.43e+4 4.43e+4 β 4.43e+4 4.43e+4 ... 4.43e+4 4.43e+4 β ... β 4.44e+4 4.44e+4 ... 4.44e+4 4.44e+4 β 4.44e+4 4.44e+4 ... 4.44e+4 4.44e+4 ββ - ββaxis 2β 4.48e+4 4.48e+4 ... 4.48e+4 4.48e+4 β 4.48e+4 4.48e+4 ... 4.48e+4 4.48e+4 β β 4.49e+4 4.49e+4 ... 4.49e+4 4.49e+4 β 4.49e+4 4.49e+4 ... 4.49e+4 4.49e+4 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 4.88e+4 4.88e+4 ... 4.88e+4 4.88e+4 β 4.88e+4 4.88e+4 ... 4.89e+4 4.89e+4 β β 4.89e+4 4.89e+4 ... 4.89e+4 4.89e+4 β 4.89e+4 4.89e+4 ... 4.89e+4 4.89e+4 ββ - ββ β 4.93e+4 4.93e+4 ... 4.93e+4 4.94e+4 β 4.94e+4 4.94e+4 ... 4.94e+4 4.94e+4 β β 4.94e+4 4.94e+4 ... 4.94e+4 4.94e+4 β 4.94e+4 4.94e+4 ... 4.94e+4 4.94e+4 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ9 @ 1 β 4.98e+4 4.98e+4 ... 4.99e+4 4.99e+4 β 4.99e+4 4.99e+4 ... 4.99e+4 4.99e+4 β ... β 4.99e+4 4.99e+4 ... 4.99e+4 4.99e+4 β 4.99e+4 4.99e+4 ... 4.99e+4 4.99e+4 ββ - ββaxis 2β 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 β 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 β β 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 β 5.04e+4 5.04e+4 ... 5.04e+4 5.04e+4 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 5.44e+4 5.44e+4 ... 5.44e+4 5.44e+4 β 5.44e+4 5.44e+4 ... 5.44e+4 5.44e+4 β β 5.44e+4 5.44e+4 ... 5.44e+4 5.44e+4 β 5.44e+4 5.44e+4 ... 5.45e+4 5.45e+4 ββ - ββ β 5.49e+4 5.49e+4 ... 5.49e+4 5.49e+4 β 5.49e+4 5.49e+4 ... 5.49e+4 5.49e+4 β β 5.49e+4 5.49e+4 ... 5.49e+4 5.49e+4 β 5.49e+4 5.50e+4 ... 5.50e+4 5.50e+4 ββ - βββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββββββββββββββββββββββββββββββββββββββ - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ€ - βββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββββββββββββββββββββββββββββββββββββββ - ββ1 @ 0 β0 @ 4 β1 @ 4 β~~~~~ β6 @ 4 β7 @ 4 ββ - ββ βaxis 5 βaxis 5 βaxis 5βaxis 5 βaxis 5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ0 @ 1 β 5.54e+4 5.54e+4 ... 5.54e+4 5.54e+4 β 5.54e+4 5.54e+4 ... 5.54e+4 5.54e+4 β ... β 5.54e+4 5.54e+4 ... 5.55e+4 5.55e+4 β 5.55e+4 5.55e+4 ... 5.55e+4 5.55e+4 ββ - ββaxis 2β 5.59e+4 5.59e+4 ... 5.59e+4 5.59e+4 β 5.59e+4 5.59e+4 ... 5.59e+4 5.59e+4 β β 5.59e+4 5.59e+4 ... 5.60e+4 5.60e+4 β 5.60e+4 5.60e+4 ... 5.60e+4 5.60e+4 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 5.99e+4 5.99e+4 ... 5.99e+4 5.99e+4 β 5.99e+4 5.99e+4 ... 5.99e+4 5.99e+4 β β 6.00e+4 6.00e+4 ... 6.00e+4 6.00e+4 β 6.00e+4 6.00e+4 ... 6.00e+4 6.00e+4 ββ - ββ β 6.04e+4 6.04e+4 ... 6.04e+4 6.04e+4 β 6.04e+4 6.04e+4 ... 6.04e+4 6.04e+4 β β 6.05e+4 6.05e+4 ... 6.05e+4 6.05e+4 β 6.05e+4 6.05e+4 ... 6.05e+4 6.05e+4 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ1 @ 1 β 6.09e+4 6.09e+4 ... 6.09e+4 6.09e+4 β 6.09e+4 6.09e+4 ... 6.10e+4 6.10e+4 β ... β 6.10e+4 6.10e+4 ... 6.10e+4 6.10e+4 β 6.10e+4 6.10e+4 ... 6.10e+4 6.10e+4 ββ - ββaxis 2β 6.14e+4 6.14e+4 ... 6.14e+4 6.14e+4 β 6.14e+4 6.14e+4 ... 6.15e+4 6.15e+4 β β 6.15e+4 6.15e+4 ... 6.15e+4 6.15e+4 β 6.15e+4 6.15e+4 ... 6.15e+4 6.15e+4 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 β 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 β β 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 β 6.55e+4 6.55e+4 ... 6.55e+4 6.55e+4 ββ - ββ β 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 β 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 β β 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 β 6.60e+4 6.60e+4 ... 6.60e+4 6.60e+4 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ~~~~~ β ... β ... β ... β ... β ... ββ - ββaxis 2β β β β β ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ8 @ 1 β 9.97e+4 9.97e+4 ... 9.97e+4 9.98e+4 β 9.98e+4 9.98e+4 ... 9.98e+4 9.98e+4 β ... β 9.98e+4 9.98e+4 ... 9.98e+4 9.98e+4 β 9.98e+4 9.98e+4 ... 9.98e+4 9.98e+4 ββ - ββaxis 2β 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 β 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 β β 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 β 1.00e+5 1.00e+5 ... 1.00e+5 1.00e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 β β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 ββ - ββ β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 β β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 β 1.04e+5 1.04e+5 ... 1.04e+5 1.04e+5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ9 @ 1 β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 β ... β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 ββ - ββaxis 2β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 β β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 β 1.05e+5 1.05e+5 ... 1.05e+5 1.05e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 β 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 β β 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 β 1.09e+5 1.09e+5 ... 1.09e+5 1.09e+5 ββ - ββ β 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 β 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 β β 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 β 1.10e+5 1.10e+5 ... 1.10e+5 1.10e+5 ββ - βββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββββββββββββββββββββββββββββββββββββββ - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ€ - β ... β - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ€ - βββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββββββββββββββββββββββββββββββββββββββ - ββ4 @ 0 β0 @ 4 β1 @ 4 β~~~~~ β6 @ 4 β7 @ 4 ββ - ββ βaxis 5 βaxis 5 βaxis 5βaxis 5 βaxis 5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ0 @ 1 β 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 β 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 β ... β 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 β 2.21e+5 2.21e+5 ... 2.21e+5 2.21e+5 ββ - ββaxis 2β 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 β 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 β β 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 β 2.22e+5 2.22e+5 ... 2.22e+5 2.22e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 β β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 ββ - ββ β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 β β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 β 2.26e+5 2.26e+5 ... 2.26e+5 2.26e+5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ1 @ 1 β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 β ... β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 ββ - ββaxis 2β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 β β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 β 2.27e+5 2.27e+5 ... 2.27e+5 2.27e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 β 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 β β 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 β 2.31e+5 2.31e+5 ... 2.31e+5 2.31e+5 ββ - ββ β 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 β 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 β β 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 β 2.32e+5 2.32e+5 ... 2.32e+5 2.32e+5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ~~~~~ β ... β ... β ... β ... β ... ββ - ββaxis 2β β β β β ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ8 @ 1 β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 β ... β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 ββ - ββaxis 2β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 β β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 β 2.66e+5 2.66e+5 ... 2.66e+5 2.66e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 β 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 β β 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 β 2.70e+5 2.70e+5 ... 2.70e+5 2.70e+5 ββ - ββ β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 β β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ9 @ 1 β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 β ... β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 β 2.71e+5 2.71e+5 ... 2.71e+5 2.71e+5 ββ - ββaxis 2β 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 β 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 β β 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 β 2.72e+5 2.72e+5 ... 2.72e+5 2.72e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 β β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 ββ - ββ β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 β β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 β 2.76e+5 2.76e+5 ... 2.76e+5 2.76e+5 ββ - βββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββββββββββββββββββββββββββββββββββββββ - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ€ - βββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββββββββββββββββββββββββββββββββββββββ - ββ5 @ 0 β0 @ 4 β1 @ 4 β~~~~~ β6 @ 4 β7 @ 4 ββ - ββ βaxis 5 βaxis 5 βaxis 5βaxis 5 βaxis 5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ0 @ 1 β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 β ... β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 ββ - ββaxis 2β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 β β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 β 2.77e+5 2.77e+5 ... 2.77e+5 2.77e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 β 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 β β 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 β 2.81e+5 2.81e+5 ... 2.81e+5 2.81e+5 ββ - ββ β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 β β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ1 @ 1 β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 β ... β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 β 2.82e+5 2.82e+5 ... 2.82e+5 2.82e+5 ββ - ββaxis 2β 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 β 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 β β 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 β 2.83e+5 2.83e+5 ... 2.83e+5 2.83e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 β β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 ββ - ββ β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 β β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 β 2.87e+5 2.87e+5 ... 2.87e+5 2.87e+5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ~~~~~ β ... β ... β ... β ... β ... ββ - ββaxis 2β β β β β ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ8 @ 1 β 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 β 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 β ... β 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 β 3.21e+5 3.21e+5 ... 3.21e+5 3.21e+5 ββ - ββaxis 2β 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 β 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 β β 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 β 3.22e+5 3.22e+5 ... 3.22e+5 3.22e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 β β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 ββ - ββ β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 β β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 β 3.26e+5 3.26e+5 ... 3.26e+5 3.26e+5 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌβββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββ9 @ 1 β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 β ... β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 ββ - ββaxis 2β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 β β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 β 3.27e+5 3.27e+5 ... 3.27e+5 3.27e+5 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 β 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 β β 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 β 3.31e+5 3.31e+5 ... 3.31e+5 3.31e+5 ββ - ββ β 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 β 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 β β 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 β 3.32e+5 3.32e+5 ... 3.32e+5 3.32e+5 ββ - βββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββββββββββββββββββββββββββββββββββββββ - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ - |}]; + {| [0]: r6x10x11x7x8x9 shape 0:6|3:7,4:8,5:9->1:10,2:11 <virtual> |}]; Tensor.print ~with_code:false ~with_grad:false `Default hoo; (* Disable line wrapping for viewing the output. In VSCode: `View: Toggle Word Wrap`. *) [%expect File "test/einsum_trivia.ml", line 1, characters 0-0: /usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum_trivia.ml _build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/einsum_trivia.ml.corrected diff --git a/_build/default/test/einsum_trivia.ml b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/default/test/einsum_trivia.ml.corrected index 35a36fa..ea67f86 100644 --- a/_build/default/test/einsum_trivia.ml +++ b/_build/.sandbox/f018ead6d62af87c68db42e044a51bc0/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 {| @@ -364,28 +239,10 @@ let%expect_test "einsum outer product" = Train.forward_and_forget backend ctx c; Tensor.print ~with_code:false ~with_grad:false `Default @@ a; [%expect - {| - ββββββββββββββββββββ - β[0]: r2 shape 0:2 β - βββ¬βββββββββββββ β - βββaxis 0 β β - βββΌβββββββββββββ€ β - βββ 0.00 1.00 β β - βββ΄βββββββββββββ β - ββββββββββββββββββββ - |}]; + {| [0]: r2 shape 0:2 <virtual> |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ b; [%expect - {| - βββββββββββββββββββββββ - β[1]: r3 shape 0:3 β - βββ¬ββββββββββββββββββββ - βββaxis 0 ββ - βββΌβββββββββββββββββββ€β - βββ 0.00 1.00 2.00 ββ - βββ΄ββββββββββββββββββββ - βββββββββββββββββββββββ - |}]; + {| [1]: r3 shape 0:3 <virtual> |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ c; [%expect {| @@ -406,37 +263,10 @@ let%expect_test "einsum outer product" = Train.forward_and_forget backend ctx c; Tensor.print ~with_code:false ~with_grad:false `Default @@ a; [%expect - {| - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ - β[5]: r2x4x3 shape 0:2|2:3->1:4 β - βββββββββ¬βββββββββββββββββββββββββ¬βββββββββββββββββββββββββββββ - ββ β0 @ 0 β1 @ 0 ββ - ββ βaxis 2 βaxis 2 ββ - βββββββββΌβββββββββββββββββββββββββΌββββββββββββββββββββββββββββ€β - ββaxis 1β 0.00 1.00 2.00 β 1.20e+1 1.30e+1 1.40e+1 ββ - ββ β 3.00 4.00 5.00 β 1.50e+1 1.60e+1 1.70e+1 ββ - ββ β 6.00 7.00 8.00 β 1.80e+1 1.90e+1 2.00e+1 ββ - ββ β 9.00 1.00e+1 1.10e+1 β 2.10e+1 2.20e+1 2.30e+1 ββ - βββββββββ΄βββββββββββββββββββββββββ΄βββββββββββββββββββββββββββββ - βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ - |}]; + {| [5]: r2x4x3 shape 0:2|2:3->1:4 <virtual> |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ b; [%expect - {| - ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ - β[6]: r5x7x6 shape 0:5|2:6->1:7 β - βββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬ββββββββββββββββββββββββββββββββββββββββββ¬βββββββββββββββββββββββββββββββββββββββββββ - ββ β0 @ 0 β1 @ 0 β2 @ 0 β3 @ 0 β4 @ 0 ββ - ββ βaxis 2 βaxis 2 βaxis 2 βaxis 2 βaxis 2 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββ€β - ββaxis 1β 0.00 1.00 ... 4.00 5.00 β 4.20e+1 4.30e+1 ... 4.60e+1 4.70e+1 β 8.40e+1 8.50e+1 ... 8.80e+1 8.90e+1 β 1.26e+2 1.27e+2 ... 1.30e+2 1.31e+2 β 1.68e+2 1.69e+2 ... 1.72e+2 1.73e+2 ββ - ββ β 6.00 7.00 ... 1.00e+1 1.10e+1 β 4.80e+1 4.90e+1 ... 5.20e+1 5.30e+1 β 9.00e+1 9.10e+1 ... 9.40e+1 9.50e+1 β 1.32e+2 1.33e+2 ... 1.36e+2 1.37e+2 β 1.74e+2 1.75e+2 ... 1.78e+2 1.79e+2 ββ - ββ β ... ... ... ... ... β ... ... ... ... ... β ... ... ... ... ... β ... ... ... ... ... β ... ... ... ... ... ββ - ββ β 3.00e+1 3.10e+1 ... 3.40e+1 3.50e+1 β 7.20e+1 7.30e+1 ... 7.60e+1 7.70e+1 β 1.14e+2 1.15e+2 ... 1.18e+2 1.19e+2 β 1.56e+2 1.57e+2 ... 1.60e+2 1.61e+2 β 1.98e+2 1.99e+2 ... 2.02e+2 2.03e+2 ββ - ββ β 3.60e+1 3.70e+1 ... 4.00e+1 4.10e+1 β 7.80e+1 7.90e+1 ... 8.20e+1 8.30e+1 β 1.20e+2 1.21e+2 ... 1.24e+2 1.25e+2 β 1.62e+2 1.63e+2 ... 1.66e+2 1.67e+2 β 2.04e+2 2.05e+2 ... 2.08e+2 2.09e+2 ββ - βββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄ββββββββββββββββββββββββββββββββββββββββββ΄βββββββββββββββββββββββββββββββββββββββββββ - ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ - |}]; + {| [6]: r5x7x6 shape 0:5|2:6->1:7 <virtual> |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ c; [%expect {| @@ -723,20 +553,7 @@ let%expect_test "einsum1 broadcast or sum out prefix axes" = let ctx = Train.forward_and_ctx backend ctx ho in 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 {| @@ -776,106 +593,7 @@ let%expect_test "einsum1 broadcast or sum out prefix axes" = let ctx = Train.forward_and_ctx backend ctx ho3 in Tensor.print ~with_code:false ~with_grad:false `Default @@ hey2; [%expect - {| - ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ - β[3]: 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 ββ - βββββββββΌββββββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββββββΌββββββββββββββββββββββββββββββββββββββββββββββ€β ...TRUNCATED BY DUNE... Tensor.print ~with_code:false ~with_grad:false `Default @@ e; [%expect - {| - βββββββββββββββββββββββββββββββββββββββββ - β[4]: r3x4 shape 1:4->0:3 β - βββββββββ¬ββββββββββββββββββββββββββββββββ - ββ βaxis 1 ββ - βββββββββΌβββββββββββββββββββββββββββββββ€β - ββaxis 0β 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 ββ - βββββββββ΄ββββββββββββββββββββββββββββββββ - βββββββββββββββββββββββββββββββββββββββββ - |}]; + {| [4]: r3x4 shape 1:4->0:3 <virtual> |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ f; [%expect {| @@ -1253,20 +888,7 @@ let%expect_test "einsum1 fixed dim axis" = let ctx = Train.forward_and_ctx backend ctx ho in 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 {| @@ -1284,20 +906,7 @@ let%expect_test "einsum1 fixed dim axis" = let ctx = Train.forward_and_ctx backend ctx ho2 in 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 @@ ho2; [%expect {| @@ -1316,18 +925,7 @@ let%expect_test "einsum1 fixed dim axis" = 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 β β - βββββββββ΄βββββββββββββ β - βββββββββββββββββββββββββββ - |}]; + {| [3]: r3x2 shape 1:2->0:3 <virtual> |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ ho3; [%expect {| @@ -1379,34 +977,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 +1024,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 includes NaNβ + βββββββββ¬βββββββββββββββββββββββββββββββ β + ββ0 @ 0 βaxis 3 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ0 @ 1 β -nan 1.79e+28 5.27e+23 β β + ββaxis 2β 2.93e+29 1.85e+25 5.27e+23 β β + ββ β ... ... ... β β + ββ β 2.93e+29 1.85e+25 5.27e+23 β β + ββ β 2.93e+29 1.06e+27 5.27e+23 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ1 @ 1 β -nan 1.79e+28 1.35e+28 β β + ββaxis 2β 3.43e-6 1.85e+25 1.35e+28 β β + ββ β ... ... ... β β + ββ β 3.43e-6 1.85e+25 1.35e+28 β β + ββ β 3.43e-6 1.06e+27 1.35e+28 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ2 @ 1 β -nan 1.79e+28 2.63e+23 β β + ββaxis 2β 4.67e+33 1.85e+25 2.63e+23 β β + ββ β ... ... ... β β + ββ β 4.67e+33 1.85e+25 2.63e+23 β β + ββ β 4.67e+33 1.06e+27 2.63e+23 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ3 @ 1 β -nan 1.79e+28 1.38e+31 β β + ββaxis 2β 1.89e+28 1.85e+25 1.38e+31 β β + ββ β ... ... ... β β + ββ β 1.89e+28 1.85e+25 1.38e+31 β β + ββ β 1.89e+28 1.06e+27 1.38e+31 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ4 @ 1 β -nan 1.79e+28 2.63e+23 β β + ββaxis 2β 9.16e-41 1.85e+25 2.63e+23 β β + ββ β ... ... ... β β + ββ β 1.55e-43 1.85e+25 2.63e+23 β β + ββ β 3.10e-41 1.06e+27 2.63e+23 β β + βββββββββ΄βββββββββββββββββββββββββββββββ β + βββββββββββββββββββββββββββββββββββββββββββββββ€ + βββββββββ¬βββββββββββββββββββββββββββββββ β + ββ1 @ 0 βaxis 3 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ0 @ 1 β -nan 2.89e+32 1.35e+28 β β + ββaxis 2β 2.93e+29 2.89e+32 1.35e+28 β β + ββ β ... ... ... β β + ββ β 2.93e+29 2.89e+32 1.35e+28 β β + ββ β 2.93e+29 2.89e+32 1.35e+28 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ1 @ 1 β -nan 2.89e+32 2.71e+28 β β + ββaxis 2β 1.00 2.89e+32 2.71e+28 β β + ββ β ... ... ... β β + ββ β 1.00 2.89e+32 2.71e+28 β β + ββ β 1.00 2.89e+32 2.71e+28 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ2 @ 1 β -nan 2.89e+32 1.35e+28 β β + ββaxis 2β 4.67e+33 2.89e+32 1.35e+28 β β + ββ β ... ... ... β β + ββ β 4.67e+33 2.89e+32 1.35e+28 β β + ββ β 4.67e+33 2.89e+32 1.35e+28 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ3 @ 1 β -nan 2.89e+32 1.38e+31 β β + ββaxis 2β 1.89e+28 2.89e+32 1.38e+31 β β + ββ β ... ... ... β β + ββ β 1.89e+28 2.89e+32 1.38e+31 β β + ββ β 1.89e+28 2.89e+32 1.38e+31 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ4 @ 1 β -nan 2.89e+32 1.35e+28 β β + ββaxis 2β 1.00 2.89e+32 1.35e+28 β β + ββ β ... ... ... β β + ββ β 1.00 2.89e+32 1.35e+28 β β + ββ β 1.00 2.89e+32 1.35e+28 β β + βββββββββ΄βββββββββββββββββββββββββββββββ β + βββββββββββββββββββββββββββββββββββββββββββββββ€ + βββββββββ¬βββββββββββββββββββββββββββββββ β + ββ2 @ 0 βaxis 3 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ0 @ 1 β -nan 1.79e+28 2.63e+23 β β + ββaxis 2β 2.93e+29 3.27e+21 2.63e+23 β β + ββ β ... ... ... β β + ββ β 2.93e+29 3.27e+21 2.63e+23 β β + ββ β 2.93e+29 1.05e+27 2.63e+23 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ1 @ 1 β -nan 1.79e+28 1.35e+28 β β + ββaxis 2β 2.00 3.27e+21 1.35e+28 β β + ββ β ... ... ... β β + ββ β 2.00 3.27e+21 1.35e+28 β β + ββ β 2.00 1.05e+27 1.35e+28 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ2 @ 1 β -nan 1.79e+28 3.39e-7 β β + ββaxis 2β 4.67e+33 3.27e+21 1.00 β β + ββ β ... ... ... β β + ββ β 4.67e+33 3.27e+21 4.00 β β + ββ β 4.67e+33 1.05e+27 5.00 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ3 @ 1 β -nan 1.79e+28 1.38e+31 β β + ββaxis 2β 1.89e+28 3.27e+21 1.38e+31 β β + ββ β ... ... ... β β + ββ β 1.89e+28 3.27e+21 1.38e+31 β β + ββ β 1.89e+28 1.05e+27 1.38e+31 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ4 @ 1 β -nan 1.79e+28 1.69e-7 β β + ββaxis 2β 2.00 3.27e+21 1.00 β β + ββ β ... ... ... β β + ββ β 2.00 3.27e+21 4.00 β β + ββ β 2.00 1.05e+27 5.00 β β + βββββββββ΄βββββββββββββββββββββββββββββββ β + βββββββββββββββββββββββββββββββββββββββββββββββ€ + βββββββββ¬βββββββββββββββββββββββββββββββ β + ββ3 @ 0 βaxis 3 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ0 @ 1 β -nan 1.79e+28 1.38e+31 β β + ββaxis 2β 2.93e+29 1.64e+19 1.38e+31 β β + ββ β ... ... ... β β + ββ β 2.93e+29 1.64e+19 1.38e+31 β β + ββ β 2.93e+29 1.05e+27 1.38e+31 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ1 @ 1 β -nan 1.79e+28 1.38e+31 β β + ββaxis 2β 3.00 1.64e+19 1.38e+31 β β + ββ β ... ... ... β β + ββ β 3.00 1.64e+19 1.38e+31 β β + ββ β 3.00 1.05e+27 1.38e+31 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ2 @ 1 β -nan 1.79e+28 1.38e+31 β β + ββaxis 2β 4.67e+33 1.64e+19 1.38e+31 β β + ββ β ... ... ... β β + ββ β 4.67e+33 1.64e+19 1.38e+31 β β + ββ β 4.67e+33 1.05e+27 1.38e+31 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ3 @ 1 β -nan 1.79e+28 2.76e+31 β β + ββaxis 2β 1.89e+28 1.64e+19 2.76e+31 β β + ββ β ... ... ... β β + ββ β 1.89e+28 1.64e+19 2.76e+31 β β + ββ β 1.89e+28 1.05e+27 2.76e+31 β β + βββββββββΌβββββββββββββββββββββββββββββββ€ β + ββ4 @ 1 β -nan 1.79e+28 1.38e+31 β β + ββaxis 2β 3.00 1.64e+19 1.38e+31 β β + ββ β ... ... ... β β + ββ β 3.00 1.64e+19 1.38e+31 β β + ββ β 3.00 1.05e+27 1.38e+31 β β + βββββββββ΄βββββββββββββββββββββββββββββββ β + βββββββββββββββββββββββββββββββββββββββββββββββ |}]; 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 1.85e+25 2.63e+23 ββ + ββ β 1.00 2.89e+32 1.35e+28 ββ + ββ β 2.00 3.27e+21 1.69e-7 ββ + ββ β 3.00 1.64e+19 1.38e+31 ββ + βββββββββ΄ββββββββββββββββββββββββββββ + βββββββββββββββββββββββββββββββββββββ |}]; 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 includes NaNβ + βββββββββ¬βββββββββββββββββββββββββββ β + ββ βaxis 1 β β + βββββββββΌβββββββββββββββββββββββββββ€ β + ββaxis 0β -nan 1.79e+28 0.00 β β + ββ β 9.14e-41 0.00 1.00 β β + ββ β ... ... ... β β + ββ β 0.00 0.00 4.00 β β + ββ β 3.08e-41 1.05e+27 5.00 β β + βββββββββ΄βββββββββββββββββββββββββββ β + ββββββββββββββββββββββββββββββββββββββββ |}] let%expect_test "einsum with a leftmost input axis preserved as output axis" = @@ -1644,54 +1218,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 +1258,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-07-02 20:06.55: Job failed: Failed: Build failed