2025-07-05 17:55.53: New job: test ahrefs/ocannl https://github.com/ahrefs/ocannl.git#refs/heads/master (04aebe83f689f400053156a388e1fd5985b16d96) (linux-x86_64:debian-11-5.3_opam-2.3) Base: ocaml/opam:debian-11-ocaml-5.3@sha256:bb9ff64fc0a30e6efc4bba2533e68fcfe1fdcbcca0b32765c1a0a4dc2592c933 Opam project build To reproduce locally: git clone --recursive "https://github.com/ahrefs/ocannl.git" -b "master" && cd "ocannl" && git reset --hard 04aebe83 cat > Dockerfile <<'END-OF-DOCKERFILE' FROM ocaml/opam:debian-11-ocaml-5.3@sha256:bb9ff64fc0a30e6efc4bba2533e68fcfe1fdcbcca0b32765c1a0a4dc2592c933 # debian-11-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 382e02b4c17b5b59d503254784ac3a5d78916d5a || git fetch origin master) && git reset -q --hard 382e02b4c17b5b59d503254784ac3a5d78916d5a && 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.13.2 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-05 17:55.53: Using cache hint "ahrefs/ocannl-ocaml/opam:debian-11-ocaml-5.3@sha256:bb9ff64fc0a30e6efc4bba2533e68fcfe1fdcbcca0b32765c1a0a4dc2592c933-debian-11-5.3_opam-2.3-e7b66700cdf32b3cc2b84269e6d26c23" 2025-07-05 17:55.53: Using OBuilder spec: ((from ocaml/opam:debian-11-ocaml-5.3@sha256:bb9ff64fc0a30e6efc4bba2533e68fcfe1fdcbcca0b32765c1a0a4dc2592c933) (comment debian-11-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 382e02b4c17b5b59d503254784ac3a5d78916d5a || git fetch origin master) && git reset -q --hard 382e02b4c17b5b59d503254784ac3a5d78916d5a && 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.13.2 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-05 17:55.53: Waiting for resource in pool OCluster 2025-07-05 17:55.53: Waiting for worker… 2025-07-05 17:55.53: Got resource from pool OCluster Building on asteria.caelum.ci.dev HEAD is now at f24a1e67 Cleanup to the `param` interface HEAD is now at 04aebe83 In progress: preparations for threefry, get rid of File_mapped (from ocaml/opam:debian-11-ocaml-5.3@sha256:bb9ff64fc0a30e6efc4bba2533e68fcfe1fdcbcca0b32765c1a0a4dc2592c933) 2025-07-05 17:55.55 ---> using "923794cf071481968755bca5f8fb939ba638f0e5adc3039ce330be9e7e13b9de" from cache /: (comment debian-11-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-05 17:55.55 ---> using "170fade88012c85b9a710b53709a9775bf94c3f978b133e60ac3ef87d3512ca8" 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. Continue? [y/n] y 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. Format upgrade done. <><> Updating repositories ><><><><><><><><><><><><><><><><><><><><><><><><><><> [default] Initialised 2025-07-05 17:55.55 ---> using "18c497649a69a9f9036d0ea729703fbea4d442984320fcdc15815f406a603dd5" 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-05 17:55.55 ---> using "1c30ab73bc45f2af7cdf17a48a03ef0ab954c1c579d4ac3bfe8e374742d81c44" from cache /src: (workdir /src) /src: (run (shell "sudo chown opam /src")) 2025-07-05 17:55.55 ---> using "484dd34db13045b21bb2fddcf6b271db41d8120a809c65fe2843ea8e4b71d87d" from cache /src: (run (cache (opam-archives (target /home/opam/.opam/download-cache))) (network host) (shell "cd ~/opam-repository && (git cat-file -e 382e02b4c17b5b59d503254784ac3a5d78916d5a || git fetch origin master) && git reset -q --hard 382e02b4c17b5b59d503254784ac3a5d78916d5a && git log --no-decorate -n1 --oneline && opam update -u")) From https://github.com/ocaml/opam-repository * branch master -> FETCH_HEAD bc726805e3..6dea2f9390 master -> origin/master 382e02b4c1 Merge pull request #28124 from kit-ty-kate/opam-publish-re.1.13.2 <><> 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-05 17:55.55 ---> using "f8518f6dce128fcc9cf4ccf98f82adddfac73062dd03aba1333df542a1dbc1e1" from cache /src: (copy (src neural_nets_lib.opam arrayjit.opam) (dst ./)) 2025-07-05 17:55.55 ---> saved as "a5f653e64e0b454028efd34658abf3ff3c93e66362466f485050ac2d6962e92c" /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-05 17:55.57 ---> saved as "7102124b4ab84c05ef52b4110b76308f88e14ec68be62de40e59222088a18cf4" /src: (run (network host) (shell "echo '(lang dune 3.0)' > './dune-project'")) 2025-07-05 17:55.57 ---> saved as "fe5081cd5e36755ac38834219d428e6cc035cdde06b46868ebc71a3bf714b58f" /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.13.2 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 "apt-get" "update" - Hit:1 http://deb.debian.org/debian bullseye InRelease - Get:2 http://deb.debian.org/debian-security bullseye-security InRelease [27.2 kB] - Get:3 http://deb.debian.org/debian bullseye-updates InRelease [44.0 kB] - Get:4 http://deb.debian.org/debian-security bullseye-security/main amd64 Packages [384 kB] - Fetched 455 kB in 1s (510 kB/s) - Reading package lists... - <><> 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: libcurl4-gnutls-dev libffi-dev pkg-config zlib1g-dev <><> Handling external dependencies <><><><><><><><><><><><><><><><><><><><><><> + /usr/bin/sudo "apt-get" "install" "-qq" "-yy" "libcurl4-gnutls-dev" "libffi-dev" "pkg-config" "zlib1g-dev" - debconf: delaying package configuration, since apt-utils is not installed - (Reading database ... (Reading database ... 5% (Reading database ... 10% (Reading database ... 15% (Reading database ... 20% (Reading database ... 25% (Reading database ... 30% (Reading database ... 35% (Reading database ... 40% (Reading database ... 45% (Reading database ... 50% (Reading database ... 55% (Reading database ... 60% (Reading database ... 65% (Reading database ... 70% (Reading database ... 75% (Reading database ... 80% (Reading database ... 85% (Reading database ... 90% (Reading database ... 95% (Reading database ... 100% (Reading database ... 18367 files and directories currently installed.) - Preparing to unpack .../00-libcurl3-gnutls_7.74.0-1.3+deb11u15_amd64.deb ... - Unpacking libcurl3-gnutls:amd64 (7.74.0-1.3+deb11u15) over (7.74.0-1.3+deb11u14) ... - Selecting previously unselected package libcurl4-gnutls-dev:amd64. - Preparing to unpack .../01-libcurl4-gnutls-dev_7.74.0-1.3+deb11u15_amd64.deb ... - Unpacking libcurl4-gnutls-dev:amd64 (7.74.0-1.3+deb11u15) ... - Selecting previously unselected package libffi-dev:amd64. - Preparing to unpack .../02-libffi-dev_3.3-6_amd64.deb ... - Unpacking libffi-dev:amd64 (3.3-6) ... - Selecting previously unselected package libglib2.0-0:amd64. - Preparing to unpack .../03-libglib2.0-0_2.66.8-1+deb11u6_amd64.deb ... - Unpacking libglib2.0-0:amd64 (2.66.8-1+deb11u6) ... - Selecting previously unselected package libglib2.0-data. - Preparing to unpack .../04-libglib2.0-data_2.66.8-1+deb11u6_all.deb ... - Unpacking libglib2.0-data (2.66.8-1+deb11u6) ... - Selecting previously unselected package libicu67:amd64. - Preparing to unpack .../05-libicu67_67.1-7+deb11u1_amd64.deb ... - Unpacking libicu67:amd64 (67.1-7+deb11u1) ... - Selecting previously unselected package libxml2:amd64. - Preparing to unpack .../06-libxml2_2.9.10+dfsg-6.7+deb11u7_amd64.deb ... - Unpacking libxml2:amd64 (2.9.10+dfsg-6.7+deb11u7) ... - Selecting previously unselected package pkg-config. - Preparing to unpack .../07-pkg-config_0.29.2-1_amd64.deb ... - Unpacking pkg-config (0.29.2-1) ... - Selecting previously unselected package shared-mime-info. - Preparing to unpack .../08-shared-mime-info_2.0-1_amd64.deb ... - Unpacking shared-mime-info (2.0-1) ... - Selecting previously unselected package xdg-user-dirs. - Preparing to unpack .../09-xdg-user-dirs_0.17-2_amd64.deb ... - Unpacking xdg-user-dirs (0.17-2) ... - Selecting previously unselected package zlib1g-dev:amd64. - Preparing to unpack .../10-zlib1g-dev_1%3a1.2.11.dfsg-2+deb11u2_amd64.deb ... - Unpacking zlib1g-dev:amd64 (1:1.2.11.dfsg-2+deb11u2) ... - Setting up libicu67:amd64 (67.1-7+deb11u1) ... - Setting up xdg-user-dirs (0.17-2) ... - Setting up libglib2.0-0:amd64 (2.66.8-1+deb11u6) ... - No schema files found: doing nothing. - Setting up libcurl3-gnutls:amd64 (7.74.0-1.3+deb11u15) ... - Setting up libcurl4-gnutls-dev:amd64 (7.74.0-1.3+deb11u15) ... - Setting up libffi-dev:amd64 (3.3-6) ... - Setting up libglib2.0-data (2.66.8-1+deb11u6) ... - Setting up pkg-config (0.29.2-1) ... - Setting up zlib1g-dev:amd64 (1:1.2.11.dfsg-2+deb11u2) ... - Setting up libxml2:amd64 (2.9.10+dfsg-6.7+deb11u7) ... - Setting up shared-mime-info (2.0-1) ... - Processing triggers for libc-bin (2.31-13+deb11u13) ... 2025-07-05 17:56.09 ---> saved as "4ad1e04768d6ec86d00f9a3535d9534cfedb2301235b3e18cfc4eb6c18aafcf7" /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.13.2 - 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 ppx_compare.v0.17.0 (cached) -> retrieved ppx_derivers.1.2.1 (cached) -> retrieved ppx_enumerate.v0.17.0 (cached) -> retrieved ppx_deriving.6.1.0 (cached) -> retrieved dune.3.19.1, dune-configurator.3.19.1 (cached) -> retrieved ppx_expect.v0.17.3 (cached) -> retrieved ppx_fields_conv.v0.17.0 (cached) -> retrieved ppx_globalize.v0.17.2 (cached) -> retrieved ppx_hash.v0.17.0 (cached) -> retrieved ppx_here.v0.17.0 (cached) -> retrieved ppx_inline_test.v0.17.1 (cached) -> retrieved ppx_optcomp.v0.17.1 (cached) -> retrieved ppx_sexp_conv.v0.17.1 (cached) -> retrieved ppx_string.v0.17.0 (cached) -> retrieved ppx_variants_conv.v0.17.1 (cached) -> installed cmdliner.1.3.0 -> installed num.1.6 -> retrieved ppx_minidebug.2.3.0 (cached) -> retrieved ppxlib_jane.v0.17.4 (cached) -> retrieved ptime.1.2.0 (cached) -> retrieved ppxlib.0.36.0 (cached) -> retrieved re.1.13.2 (cached) -> retrieved result.1.5 (cached) -> retrieved seq.base (cached) -> installed seq.base -> retrieved sexplib.v0.17.0 (cached) -> retrieved printbox.0.12, printbox-ext-plot.0.12, printbox-html.0.12, printbox-md.0.12, printbox-text.0.12 (cached) -> retrieved saturn_lockfree.0.5.0 (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.13.2 -> 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 angstrom.0.16.1 -> installed tyxml.4.6.0 -> installed printbox-html.0.12 -> installed ctypes.0.23.0 -> installed base.v0.17.3 -> installed variantslib.v0.17.0 -> installed fieldslib.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 curl.0.10.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-05 17:57.09 ---> saved as "3eb899ee86abc63879ffb5317c711c0129b5fcee03b14e338f9507564652c806" /src: (copy (src .) (dst /src)) 2025-07-05 17:57.09 ---> saved as "e69c151f7296d79e8d2fc5dbbada9f3d96c9290119ab588da7e294eb911030fa" /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_ppx/test_ppx_op_expected.ml", line 1, characters 0-0: /usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test_ppx/test_ppx_op_expected.ml _build/default/test_ppx/test_ppx_op_actual.ml diff --git a/_build/default/test_ppx/test_ppx_op_expected.ml b/_build/default/test_ppx/test_ppx_op_actual.ml index a6aa4ae..c971ef0 100644 --- a/_build/default/test_ppx/test_ppx_op_expected.ml +++ b/_build/default/test_ppx/test_ppx_op_actual.ml @@ -2,20 +2,20 @@ open Base open Ocannl module TDSL = Operation.TDSL let y0 = - let hey1 = TDSL.param ?values:None "hey1" in + let hey1 = TDSL.param ?value:None "hey1" in let open! TDSL.O in ((+) ?label:(Some ["y0"])) ((( *. ) ?label:None) (TDSL.number (Float.of_int 2)) hey1) (TDSL.number (Float.of_int 3)) let y1 = - let hey2 = TDSL.param ?values:None "hey2" in + let hey2 = TDSL.param ?value:None "hey2" in let open! TDSL.O in fun x -> ((+) ?label:(Some (List.concat [["y1"]; (x.Tensor.value).Ir.Tnode.label]))) ((( * ) ?label:None) hey2 (TDSL.number (Float.of_int 2))) x let y2 = - let hey3 = TDSL.param ?values:None "hey3" in + let hey3 = TDSL.param ?value:None "hey3" in let open! TDSL.O in fun x1 x2 -> ((+) ?label:(Some @@ -36,14 +36,14 @@ let b = ~output_dims:[2] [|(Float.of_int 7);(Float.of_int 8);(Float.of_int 9);(Float.of_int 10)|] let y = - let hey4 = TDSL.param ?values:None "hey4" in + let hey4 = TDSL.param ?value:None "hey4" in let open! TDSL.O in ((+) ?label:(Some ["y"])) ((( * ) ?label:None) hey4 (TDSL.number ?label:None ~axis_label:"q" 2.0)) (TDSL.number ?label:None ~axis_label:"p" 1.0) let z = - let hey5 = TDSL.param ?values:None "hey5" - and hey6 = TDSL.param ?values:None "hey6" in + let hey5 = TDSL.param ?value:None "hey5" + and hey6 = TDSL.param ?value:None "hey6" in let open! TDSL.O in ((+) ?label:(Some ["z"])) ((( * ) ?label:None) (TDSL.number ?label:None ~axis_label:"q" 2.0) hey5) @@ -51,8 +51,8 @@ let z = let stride = 2 and dilation = 3 let z2 = - let hey7 = TDSL.param ?values:None "hey7" - and hey8 = TDSL.param ?values:None "hey8" in + let hey7 = TDSL.param ?value:None "hey7" + and hey8 = TDSL.param ?value:None "hey8" in let open! TDSL.O in TDSL.einsum ?label:(Some ["z2"]) (String.concat ~sep:"" @@ -61,8 +61,8 @@ let z2 = let z3 = let s = 2 and d = 3 in - let hey10 = TDSL.param ?values:None "hey10" - and hey9 = TDSL.param ?values:None "hey9" in + let hey10 = TDSL.param ?value:None "hey10" + and hey9 = TDSL.param ?value:None "hey9" in let open! TDSL.O in TDSL.einsum ?label:(Some []) (String.concat ~sep:"" @@ -78,7 +78,7 @@ let mlp_layer = let b = (TDSL.param ~more_label:(config.label)) ~output_dims:[config.hid_dim] "b" - and w = (TDSL.param ~more_label:(config.label)) ?values:None "w" in + and w = (TDSL.param ~more_label:(config.label)) ?value:None "w" in fun x -> (relu ?label:(Some 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: 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... (cd _build/default/test_ppx && ./test_ppx_op.exe) Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test_ppx/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/default/test_ppx && ./test_ppx_op_expected.exe) Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test_ppx/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/default/test && .tutorials.inline-tests/inline-test-runner.exe inline-test-runner tutorials -partition 'Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/f2737efa811bce62a8093be4ad7d0f4a/default/test/ocannl_config.' -source-tree-root .. -diff-cmd -) Welcome to OCANNL! Reading configuration defaults from /src/_build/.sandbox/159ee49257697c1d31904c383c87c004/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/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 (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/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..98fdddb 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 @@ -735,173 +735,173 @@ HERE: test/einsum/einsum_trivia_exec.ml:269:21 HERE: test/einsum/einsum_trivia_exec.ml:298:21 -┌────────────────────────────────┐ -│[9]: ;=>+ shape 0:4,1:5,2:6,3:3 │ -│┌──────┬──────────────────┐ │ -││0 @ 0 │axis 3 │ │ -│├──────┼──────────────────┤ │ -││0 @ 1 │ 0.00 0.00 0.00 │ │ -││axis 2│ 0.00 0.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 0.00 0.00 4.00 │ │ -││ │ 0.00 0.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││1 @ 1 │ 0.00 1.00 0.00 │ │ -││axis 2│ 0.00 1.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 0.00 1.00 4.00 │ │ -││ │ 0.00 1.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││2 @ 1 │ 0.00 2.00 0.00 │ │ -││axis 2│ 0.00 2.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 0.00 2.00 4.00 │ │ -││ │ 0.00 2.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││3 @ 1 │ 0.00 3.00 0.00 │ │ -││axis 2│ 0.00 3.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 0.00 3.00 4.00 │ │ -││ │ 0.00 3.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││4 @ 1 │ 0.00 4.00 0.00 │ │ -││axis 2│ 0.00 4.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 0.00 4.00 4.00 │ │ -││ │ 0.00 4.00 5.00 │ │ -│└──────┴──────────────────┘ │ -├────────────────────────────────┤ -│┌──────┬──────────────────┐ │ -││1 @ 0 │axis 3 │ │ -│├──────┼──────────────────┤ │ -││0 @ 1 │ 1.00 0.00 0.00 │ │ -││axis 2│ 1.00 0.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 1.00 0.00 4.00 │ │ -││ │ 1.00 0.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││1 @ 1 │ 1.00 1.00 0.00 │ │ -││axis 2│ 1.00 1.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 1.00 1.00 4.00 │ │ -││ │ 1.00 1.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││2 @ 1 │ 1.00 2.00 0.00 │ │ -││axis 2│ 1.00 2.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 1.00 2.00 4.00 │ │ -││ │ 1.00 2.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││3 @ 1 │ 1.00 3.00 0.00 │ │ -││axis 2│ 1.00 3.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 1.00 3.00 4.00 │ │ -││ │ 1.00 3.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││4 @ 1 │ 1.00 4.00 0.00 │ │ -││axis 2│ 1.00 4.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 1.00 4.00 4.00 │ │ -││ │ 1.00 4.00 5.00 │ │ -│└──────┴──────────────────┘ │ -├────────────────────────────────┤ -│┌──────┬──────────────────┐ │ -││2 @ 0 │axis 3 │ │ -│├──────┼──────────────────┤ │ -││0 @ 1 │ 2.00 0.00 0.00 │ │ -││axis 2│ 2.00 0.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 2.00 0.00 4.00 │ │ -││ │ 2.00 0.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││1 @ 1 │ 2.00 1.00 0.00 │ │ -││axis 2│ 2.00 1.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 2.00 1.00 4.00 │ │ -││ │ 2.00 1.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││2 @ 1 │ 2.00 2.00 0.00 │ │ -││axis 2│ 2.00 2.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 2.00 2.00 4.00 │ │ -││ │ 2.00 2.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││3 @ 1 │ 2.00 3.00 0.00 │ │ -││axis 2│ 2.00 3.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 2.00 3.00 4.00 │ │ -││ │ 2.00 3.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││4 @ 1 │ 2.00 4.00 0.00 │ │ -││axis 2│ 2.00 4.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 2.00 4.00 4.00 │ │ -││ │ 2.00 4.00 5.00 │ │ -│└──────┴──────────────────┘ │ -├────────────────────────────────┤ -│┌──────┬──────────────────┐ │ -││3 @ 0 │axis 3 │ │ -│├──────┼──────────────────┤ │ -││0 @ 1 │ 3.00 0.00 0.00 │ │ -││axis 2│ 3.00 0.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 3.00 0.00 4.00 │ │ -││ │ 3.00 0.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││1 @ 1 │ 3.00 1.00 0.00 │ │ -││axis 2│ 3.00 1.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 3.00 1.00 4.00 │ │ -││ │ 3.00 1.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││2 @ 1 │ 3.00 2.00 0.00 │ │ -││axis 2│ 3.00 2.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 3.00 2.00 4.00 │ │ -││ │ 3.00 2.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││3 @ 1 │ 3.00 3.00 0.00 │ │ -││axis 2│ 3.00 3.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 3.00 3.00 4.00 │ │ -││ │ 3.00 3.00 5.00 │ │ -│├──────┼──────────────────┤ │ -││4 @ 1 │ 3.00 4.00 0.00 │ │ -││axis 2│ 3.00 4.00 1.00 │ │ -││ │ ... ... ... │ │ -││ │ 3.00 4.00 4.00 │ │ -││ │ 3.00 4.00 5.00 │ │ -│└──────┴──────────────────┘ │ -└────────────────────────────────┘ +┌─────────────────────────────────────┐ +│[9]: ;=>+ shape 0:4,1:5,2:6,3:3 │ +│┌──────┬───────────────────────────┐ │ +││0 @ 0 │axis 3 │ │ +│├──────┼───────────────────────────┤ │ +││0 @ 1 │ 4.59e+29 3.06e-41 0.00 │ │ +││axis 2│ 4.59e+29 2.91e+3 1.00 │ │ +││ │ ... ... ... │ │ +││ │ 4.59e+29 5.12e+3 4.00 │ │ +││ │ 4.59e+29 3.06e-41 5.00 │ │ +│├──────┼───────────────────────────┤ │ +││1 @ 1 │ 0.00 1.00 2.06e+3 │ │ +││axis 2│ 0.00 2.91e+3 2.06e+3 │ │ +││ │ ... ... ... │ │ +││ │ 4.68e+3 5.12e+3 2.06e+3 │ │ +││ │ 0.00 1.00 2.06e+3 │ │ +│├──────┼───────────────────────────┤ │ +││2 @ 1 │ 2.14e+3 2.00 2.46e+3 │ │ +││axis 2│ 2.14e+3 2.91e+3 2.46e+3 │ │ +││ │ ... ... ... │ │ +││ │ 6.82e+3 5.12e+3 2.46e+3 │ │ +││ │ 2.14e+3 2.00 2.46e+3 │ │ +│├──────┼───────────────────────────┤ │ +││3 @ 1 │ 2.48e+3 3.00 2.86e+3 │ │ +││axis 2│ 2.48e+3 2.91e+3 2.86e+3 │ │ +││ │ ... ... ... │ │ +││ │ 7.16e+3 5.12e+3 2.86e+3 │ │ +││ │ 2.48e+3 3.00 2.86e+3 │ │ +│├──────┼───────────────────────────┤ │ +││4 @ 1 │ 2.82e+3 4.00 3.26e+3 │ │ +││axis 2│ 2.82e+3 2.91e+3 3.26e+3 │ │ +││ │ ... ... ... │ │ +││ │ 7.50e+3 5.12e+3 3.26e+3 │ │ +││ │ 2.82e+3 4.00 3.26e+3 │ │ +│└──────┴───────────────────────────┘ │ +├─────────────────────────────────────┤ +│┌──────┬────────────────────────────┐│ +││1 @ 0 │axis 3 ││ +│├──────┼────────────────────────────┤│ +││0 @ 1 │ 4.59e+29 4.80e+1 6.50e+1 ││ +││axis 2│ 4.59e+29 2.96e+3 6.60e+1 ││ +││ │ ... ... ... ││ +││ │ 4.59e+29 5.16e+3 6.90e+1 ││ +││ │ 4.59e+29 4.80e+1 7.00e+1 ││ +│├──────┼────────────────────────────┤│ +││1 @ 1 │ 1.00 4.90e+1 2.12e+3 ││ +││axis 2│ 1.00 2.96e+3 2.12e+3 ││ +││ │ ... ... ... ││ +││ │ 4.68e+3 5.16e+3 2.12e+3 ││ +││ │ 1.00 4.90e+1 2.13e+3 ││ +│├──────┼────────────────────────────┤│ +││2 @ 1 │ 2.14e+3 5.00e+1 2.52e+3 ││ +││axis 2│ 2.14e+3 2.96e+3 2.52e+3 ││ +││ │ ... ... ... ││ +││ │ 6.82e+3 5.17e+3 2.52e+3 ││ +││ │ 2.14e+3 5.00e+1 2.53e+3 ││ +│├──────┼────────────────────────────┤│ +││3 @ 1 │ 2.48e+3 5.10e+1 2.92e+3 ││ +││axis 2│ 2.48e+3 2.96e+3 2.92e+3 ││ +││ │ ... ... ... ││ +││ │ 7.16e+3 5.17e+3 2.92e+3 ││ +││ │ 2.48e+3 5.10e+1 2.93e+3 ││ +│├──────┼────────────────────────────┤│ +││4 @ 1 │ 2.82e+3 5.20e+1 3.32e+3 ││ +││axis 2│ 2.82e+3 2.96e+3 3.32e+3 ││ +││ │ ... ... ... ││ +││ │ 7.50e+3 5.17e+3 3.32e+3 ││ +││ │ 2.82e+3 5.20e+1 3.33e+3 ││ +│└──────┴────────────────────────────┘│ +├─────────────────────────────────────┤ +│┌──────┬────────────────────────────┐│ +││2 @ 0 │axis 3 ││ +│├──────┼────────────────────────────┤│ +││0 @ 1 │ 4.59e+29 1.05e+2 1.60e+2 ││ +││axis 2│ 4.59e+29 3.01e+3 1.61e+2 ││ +││ │ ... ... ... ││ +││ │ 4.59e+29 5.22e+3 1.64e+2 ││ +││ │ 4.59e+29 1.05e+2 1.65e+2 ││ +│├──────┼────────────────────────────┤│ +││1 @ 1 │ 2.00 1.06e+2 2.22e+3 ││ +││axis 2│ 2.00 3.01e+3 2.22e+3 ││ +││ │ ... ... ... ││ +││ │ 4.68e+3 5.22e+3 2.22e+3 ││ +││ │ 2.00 1.06e+2 2.22e+3 ││ +│├──────┼────────────────────────────┤│ +││2 @ 1 │ 2.15e+3 1.07e+2 2.62e+3 ││ +││axis 2│ 2.15e+3 3.01e+3 2.62e+3 ││ +││ │ ... ... ... ││ +││ │ 6.83e+3 5.22e+3 2.62e+3 ││ +││ │ 2.15e+3 1.07e+2 2.62e+3 ││ +│├──────┼────────────────────────────┤│ +││3 @ 1 │ 2.48e+3 1.08e+2 3.02e+3 ││ +││axis 2│ 2.48e+3 3.02e+3 3.02e+3 ││ +││ │ ... ... ... ││ +││ │ 7.16e+3 5.22e+3 3.02e+3 ││ +││ │ 2.48e+3 1.08e+2 3.02e+3 ││ +│├──────┼────────────────────────────┤│ +││4 @ 1 │ 2.82e+3 1.09e+2 3.42e+3 ││ +││axis 2│ 2.82e+3 3.02e+3 3.42e+3 ││ +││ │ ... ... ... ││ +││ │ 7.50e+3 5.22e+3 3.42e+3 ││ +││ │ 2.82e+3 1.09e+2 3.42e+3 ││ +│└──────┴────────────────────────────┘│ +├─────────────────────────────────────┤ +│┌──────┬────────────────────────────┐│ +││3 @ 0 │axis 3 ││ +│├──────┼────────────────────────────┤│ +││0 @ 1 │ 4.59e+29 2.20e+2 2.53e+2 ││ +││axis 2│ 4.59e+29 3.13e+3 2.54e+2 ││ +││ │ ... ... ... ││ +││ │ 4.59e+29 5.34e+3 2.57e+2 ││ +││ │ 4.59e+29 2.20e+2 2.58e+2 ││ +│├──────┼────────────────────────────┤│ +││1 @ 1 │ 3.00 2.21e+2 2.31e+3 ││ +││axis 2│ 3.00 3.13e+3 2.31e+3 ││ +││ │ ... ... ... ││ +││ │ 4.68e+3 5.34e+3 2.31e+3 ││ +││ │ 3.00 2.21e+2 2.31e+3 ││ +│├──────┼────────────────────────────┤│ +││2 @ 1 │ 2.15e+3 2.22e+2 2.71e+3 ││ +││axis 2│ 2.15e+3 3.13e+3 2.71e+3 ││ +││ │ ... ... ... ││ +││ │ 6.83e+3 5.34e+3 2.71e+3 ││ +││ │ 2.15e+3 2.22e+2 2.71e+3 ││ +│├──────┼────────────────────────────┤│ +││3 @ 1 │ 2.48e+3 2.23e+2 3.11e+3 ││ +││axis 2│ 2.48e+3 3.13e+3 3.11e+3 ││ +││ │ ... ... ... ││ +││ │ 7.16e+3 5.34e+3 3.11e+3 ││ +││ │ 2.48e+3 2.23e+2 3.11e+3 ││ +│├──────┼────────────────────────────┤│ +││4 @ 1 │ 2.82e+3 2.24e+2 3.51e+3 ││ +││axis 2│ 2.82e+3 3.13e+3 3.51e+3 ││ +││ │ ... ... ... ││ +││ │ 7.50e+3 5.34e+3 3.51e+3 ││ +││ │ 2.82e+3 2.24e+2 3.51e+3 ││ +│└──────┴────────────────────────────┘│ +└─────────────────────────────────────┘ HERE: test/einsum/einsum_trivia_exec.ml:299:21 -┌───────────────────────────┐ -│[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 3.06e-41 0.00 ││ +││ │ 1.00 4.80e+1 6.50e+1 ││ +││ │ 2.00 1.05e+2 1.60e+2 ││ +││ │ 3.00 2.20e+2 2.53e+2 ││ +│└──────┴─────────────────────────┘│ +└──────────────────────────────────┘ HERE: test/einsum/einsum_trivia_exec.ml:300:21 -┌───────────────────────────┐ -│[7]: =>_tk shape 0:6,1:3 │ -│┌──────┬──────────────────┐│ -││ │axis 1 ││ -│├──────┼──────────────────┤│ -││axis 0│ 0.00 0.00 0.00 ││ -││ │ 0.00 0.00 1.00 ││ -││ │ ... ... ... ││ -││ │ 0.00 0.00 4.00 ││ -││ │ 0.00 0.00 5.00 ││ -│└──────┴──────────────────┘│ -└───────────────────────────┘ +┌─────────────────────────────────┐ +│[7]: =>_tk shape 0:6,1:3 │ +│┌──────┬────────────────────────┐│ +││ │axis 1 ││ +│├──────┼────────────────────────┤│ +││axis 0│ 0.00 0.00 0.00 ││ +││ │ 0.00 2.91e+3 1.00 ││ +││ │ ... ... ... ││ +││ │ 4.68e+3 5.12e+3 4.00 ││ +││ │ 0.00 0.00 5.00 ││ +│└──────┴────────────────────────┘│ +└─────────────────────────────────┘ HERE: test/einsum/einsum_trivia_exec.ml:323:21 (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/default/test/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file File "test/einsum/moons_demo_variant.expected", line 1, characters 0-0: /usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum/moons_demo_variant.expected _build/default/test/einsum/moons_demo_variant.exe.output diff --git a/_build/default/test/einsum/moons_demo_variant.expected b/_build/default/test/einsum/moons_demo_variant.exe.output index 68171df..953eeeb 100644 --- a/_build/default/test/einsum/moons_demo_variant.expected +++ b/_build/default/test/einsum/moons_demo_variant.exe.output @@ -1,121 +1,104 @@ -Welcome to OCANNL! Reading configuration defaults from /Users/lukstafi/ocannl/_build/default/test/einsum/ocannl_config. +Welcome to OCANNL! Reading configuration defaults from /src/_build/default/test/einsum/ocannl_config. Retrieving commandline, environment, or config file variable ocannl_log_level Found 0, in the config file Half-moons scatterplot and decision boundary: ┌────────────────────────────────────────────────────────────────────────────────────────────────────┐ -│****************************#***********************************************************************│ -│****************************###***###***#***********************************************************│ -│***************************#********#***#**********************************************************.│ -│********************#*********************##*#*#************************************************....│ -│******************#***#*****##*#*#*#****#*#*#*##**********************************************......│ -│**************#*#**###**#*##**##********#**#****###**#*************************************.........│ -│***********#***##***#**##*****************#*#*##*###*************************************...........│ -│***************#**###******************************##*#********************************.............│ -│*************#*****#******************************#*#***#*#*************************................│ -│*********#***#************************************##*###**#***********************..................│ -│******#***#*#*#******************************************#*#*******************.....................│ -│*****###****#*******************************************#********************.......................│ -│*******#*###***********************************************#*#************..........................│ -│**********##**********************%%*********************#**##**********............................│ -│*****#***#**********************%**%%********************************.............................%.│ -│**####*************************%**%**************************##*#**............................%...%│ -│****#**************************%%**%*************************#*#*..............................%....│ -│*#****##************************%*...%*********************##*..#...........................%...%%..│ -│****#*#***********************.%..%.%........................#..#.............................%.%...│ -│##*##*********************.......%.............................#.##.........................%.%.%%%.│ -│**#####*****************.......%...%%.......................................................%..%.%..│ -│**********************...........%.%.%%.....................#.#.#.#.............................%%..│ -│***##****************............%..%..%........................#.#.........................%...%...│ -│##**#**************................%...%.....................#..###......................%.%%.......│ -│###***************...............%.%.%...........................##............................%....│ -│**##************.................................................#......................%%.%..%.....│ -│**#************....................%%...........................##......................%%.%........│ -│**************........................%......................#..#...........................%.......│ -│************........................%%.%.......................##.....................%..%..........│ -│***********............................%%..............................................%%.%%........│ -│**********.............................%...%............................................%%%.........│ -│********...................................%..%....................................%.....%..........│ -│*******.................................%%..%.%................................%.%.%.%%.............│ -│******.....................................%%.%...%%.%.......................%%%....%.%.%...........│ -│*****....................................%.....%.%..%....................%.....%%.%.%...............│ -│***.........................................%%...%.%%%%..%.......%.........%%.%%....................│ -│**................................................%..%.......%..%%%%...%.%..%.%.....................│ -│*..................................................%.%.%............%..%...%........................│ -│..................................................%....%.%......%%...%.%............................│ -│..........................................................%..%......%%%.............................│ +│...................................#................................................................│ +│..........................##..#..###....##..........................................................│ +│.........................#....#.....#...............................................................│ +│.......................#.##.#.......#####...#.#.....................................................│ +│..................#.#.#...#..#.##....#.#.##..#....#.................................................│ +│.............#.#......##.#..#..........###..#....#..................................................│ +│.............##..##.....#....................#.##.###...#...........................................│ +│.............#.##.#.#.#.......................#...##..#.............................................│ +│.........#.#....##..................................##..#...........................................│ +│...........#.#....................................#..##...#.#.......................................│ +│..........##..........................................#...#.#.......................................│ +│.......#...###............................................#.........................................│ +│....#....##..........................%..................#....##.....................................│ +│.........##................................................#..#..............................%......│ +│.....#..#........................%..%......................##.#...................................%.│ +│...##.#..................................................#....................................%...%.│ +│..###...........................%..%%......................###................................%.%...│ +│..#...#.#.......................%%.%%.......................#.##.#............................%.....│ +│..#...#.............................%.........................##..............................%...%.│ +│..##..#............................%%.........................#..##.........................%%..%.%%│ +│#...#..#.........................%.%%%.......................##...#............................%.%..│ +│##...#.............................%%.........................#.#..#................................│ +│...................................%%%%.......................##.#.#........................%%..%.%.│ +│###................................%%..............................#.......................%....%...│ +│..##................................%%%.%....................................................%%..%..│ +│.#.....................................%.......................#.###......................%.%%.%....│ +│.....#................................%...%................................................%...%....│ +│...................................%.....%%......................#..................................│ +│.....................................%%%..%...........................................%..%%..%......│ +│.......................................%%.....%......................................%..%.%.%.......│ +│.........................................%%............................................%..%%.%......│ +│..........................................%.%.....%...............................%.%.%%..%.........│ +│.........................................%%%..%%......................................%.............│ +│...........................................%.%..%..............................%%.%.%.%%.%..........│ +│..............................................%%%%..%.%.......................%%..%.%.%.............│ +│..............................................%..%%%.%%...................%..%%...%.................│ +│...............................................%...%.%.%..%...%.%%..%%.%%...%.......................│ +│......................................................%%.%%%..%.%..%%.%.%.%%........................│ +│.....................................................%...........%%%%......%%.......................│ +│.......................................................%.%%%%.%.......%...%.........................│ └────────────────────────────────────────────────────────────────────────────────────────────────────┘Loss: -┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ -│ 3.79e+1│- │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│l │ │ -│o │ │ -│s │ │ -│s │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ - │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ │ -│ │ - - - - │ -│ 0.00 │ - ---- --- --- ---- ---- ---- ---- -- - ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │ -├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ -│ │0.00 7.90e+1│ -│ │ step │ -└────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘Log-loss, for better visibility: +┌─────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ +│ 0.00│ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│l │ │ +│o │ │ +│s │ │ +│s │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ 0.00│---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │ +├─────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ +│ │0.00 7.90e+1│ +│ │ step │ +└─────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘Log-loss, for better visibility: ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐ -│ 3.63 │- │ +│ -1.00e+1│ │ │ │ │ │ │ │ │ │ │ -│ │ - │ │ │ │ │ │ │ │ │ │ │ │ │ -│ │ - │ -│ │ - │ -│ │ - - - │ -│ │ - - - - │ -│ │ -- - - - - - - - -- - - - - - -- │ -│ │ - - - - - - - -- - - -- - - - - │ -│ │ - - - │ -│l │ - - - -- - │ -│o │ - - │ -│g │ - - - - │ -│ │ - - - │ -│l │ - -- │ -│o │ - - - │ -│s │ - │ -│s │ - │ -│ │ - │ │ │ │ │ │ │ │ │ │ @@ -124,13 +107,30 @@ Half-moons scatterplot and decision boundary: │ │ │ │ │ │ │ │ │ +│l │ │ +│o │ │ +│g │ │ │ │ │ +│l │ │ +│o │ │ +│s │ │ +│s │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ -│ -1.00e+1│ - - - - - - - │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ │ │ +│ -1.00e+1│---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │ ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤ │ │0.00 7.90e+1│ │ │ step │ (cd _build/.sandbox/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/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/159ee49257697c1d31904c383c87c004/default/test/zero2hero_1of7.ml.corrected diff --git a/_build/default/test/zero2hero_1of7.ml b/_build/.sandbox/159ee49257697c1d31904c383c87c004/default/test/zero2hero_1of7.ml.corrected index 3b7cc00..47def64 100644 --- a/_build/default/test/zero2hero_1of7.ml +++ b/_build/.sandbox/159ee49257697c1d31904c383c87c004/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 + 7.73e+23 + #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/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/159ee49257697c1d31904c383c87c004/default/test/micrograd_demo.ml.corrected diff --git a/_build/default/test/micrograd_demo.ml b/_build/.sandbox/159ee49257697c1d31904c383c87c004/default/test/micrograd_demo.ml.corrected index b394efe..4062941 100644 --- a/_build/default/test/micrograd_demo.ml +++ b/_build/.sandbox/159ee49257697c1d31904c383c87c004/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 (); @@ -168,223 +137,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 @@ @@ -398,4 +178,20 @@ 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(\"The linked context lacks node b1\")") + 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.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 125, 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/159ee49257697c1d31904c383c87c004/default/test/hello_world_op.ml.corrected diff --git a/_build/default/test/hello_world_op.ml b/_build/.sandbox/159ee49257697c1d31904c383c87c004/default/test/hello_world_op.ml.corrected index 6407bb4..cd5311b 100644 --- a/_build/default/test/hello_world_op.ml +++ b/_build/.sandbox/159ee49257697c1d31904c383c87c004/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 │ │ + │├┼──────────┤ │ + │││ -2.94e+1 │ │ + │└┴──────────┘ │ └────────────────────┘ |}] @@ -67,11 +67,11 @@ let%expect_test "Matrix multiplication dims 1x1" = {| ┌────────────────────────┐ │[0]: hey shape 1:1->0:1 │ - │┌──────┬──────┐ │ - ││ │axis 1│ │ - │├──────┼──────┤ │ - ││axis 0│ 7.00 │ │ - │└──────┴──────┘ │ + │┌──────┬──────────┐ │ + ││ │axis 1 │ │ + │├──────┼──────────┤ │ + ││axis 0│ -1.45e+1 │ │ + │└──────┴──────────┘ │ └────────────────────────┘ |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ y; @@ -79,11 +79,11 @@ let%expect_test "Matrix multiplication dims 1x1" = {| ┌───────────────────┐ │[6]: +_y shape 0:1 │ - │┌┬─────────┐ │ - │││axis 0 │ │ - │├┼─────────┤ │ - │││ 1.50e+1 │ │ - │└┴─────────┘ │ + │┌┬──────────┐ │ + │││axis 0 │ │ + │├┼──────────┤ │ + │││ -2.80e+1 │ │ + │└┴──────────┘ │ └───────────────────┘ |}] @@ -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]: [| [ 0.00 ; 0.00 ; 0.00 ] ; [ 0.00 ; 0.00 ; 0.00 ] |]_hoo shape 0:2|1:3 [| [ 1.00 ; 2.00 ; 3.00 ] ; [ 4.00 ; 5.00 ; 6.00 ] |] @@ -140,7 +140,7 @@ let%expect_test "Print constant tensor" = [%expect {| ┌─────────────────────────────────────────────────────────────────────────────┐ - │[1]: [| [ 1.00 ; 2.00 ; 3.00 ] ; [ 4.00 ; 5.00 ; 6.00 ] |]_hoo shape 0:2|1:3 │ + │[1]: [| [ 0.00 ; 0.00 ; 0.00 ] ; [ 0.00 ; 0.00 ; 0.00 ] |]_hoo shape 0:2|1:3 │ │┌──────┬──────────────────┐ │ ││ │axis 1 │ │ │├──────┼──────────────────┤ │ @@ -471,28 +471,28 @@ let%expect_test "Matrix multiplication dims 2x3" = 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 │ │ - │└──────┴────────────┘ │ - └────────────────────────┘ + ┌─────────────────────────────┐ + │[0]: hey shape 1:2->0:3 │ + │┌──────┬────────────────────┐│ + ││ │axis 1 ││ + │├──────┼────────────────────┤│ + ││axis 0│ -1.47e+1 3.07e-41 ││ + ││ │ 0.00 0.00 ││ + ││ │ 4.48e-44 0.00 ││ + │└──────┴────────────────────┘│ + └─────────────────────────────┘ |}]; 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 ││ - │└┴───────────────────────────┘│ - └──────────────────────────────┘ + ┌─────────────────────────┐ + │[6]: +_y shape 0:3 │ + │┌┬──────────────────────┐│ + │││axis 0 ││ + │├┼──────────────────────┤│ + │││ -2.55e+1 5.00 6.00 ││ + │└┴──────────────────────┘│ + └─────────────────────────┘ |}] let%expect_test "Big matrix" = @@ -516,70 +516,37 @@ let%expect_test "Big matrix" = Train.forward_and_forget backend ctx y; Tensor.print ~with_code:false ~with_grad:false `Inline zero_to_twenty; [%expect - {| - [2]: 0...20 shape 0:21 [ - 0.00 - ; 1.00 - ; 2.00 - ; 3.00 - ; 4.00 - ; 5.00 - ; 6.00 - ; 7.00 - ; 8.00 - ; 9.00 - ; 10.00 - ; 11.00 - ; 12.00 - ; 13.00 - ; 14.00 - ; 15.00 - ; 16.00 - ; 17.00 - ; 18.00 - ; 19.00 - ; 20.00 - ] - |}]; + {| [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 │ - │┌┬───────────────────────────────────┐│ - │││axis 0 ││ - │├┼───────────────────────────────────┤│ - │││ 0.00 1.00 ... 1.90e+1 2.00e+1 ││ - │└┴───────────────────────────────────┘│ - └──────────────────────────────────────┘ - |}]; + {| [2]: 0...20 shape 0:21 <virtual> |}]; 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 ││ - │└──────┴─────────────────────────────────────────┘│ - └──────────────────────────────────────────────────┘ + ┌──────────────────────────────────────────────────────────┐ + │[0]: hey shape 1:21->0:21 │ + │┌──────┬─────────────────────────────────────────────────┐│ + ││ │axis 1 ││ + │├──────┼─────────────────────────────────────────────────┤│ + ││axis 0│ 2.62e-29 4.55e-41 ... -1.36e-38 -1.18e+21 ││ + ││ │ -2.28e-26 -1.36e-38 ... -1.36e-38 -1.18e+21 ││ + ││ │ ... ... ... ... ... ││ + ││ │ -2.28e-26 -1.36e-38 ... -1.36e-38 -1.18e+21 ││ + ││ │ -2.28e-26 -1.36e-38 ... -2.28e-26 2.50 ││ + │└──────┴─────────────────────────────────────────────────┘│ + └──────────────────────────────────────────────────────────┘ |}]; 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 ││ - │└┴─────────────────────────────────────────┘│ - └────────────────────────────────────────────┘ + ┌────────────────────────────────────────────────────┐ + │[5]: + shape 0:21 │ + │┌┬─────────────────────────────────────────────────┐│ + │││axis 0 ││ + │├┼─────────────────────────────────────────────────┤│ + │││ -8.30e+22 -9.13e+22 ... -9.70e+22 -8.30e+21 ││ + │└┴─────────────────────────────────────────────────┘│ + └────────────────────────────────────────────────────┘ |}] let%expect_test "Very big tensor" = @@ -603,140 +570,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/159ee49257697c1d31904c383c87c004/default/test/einsum_trivia.ml.corrected diff --git a/_build/default/test/einsum_trivia.ml b/_build/.sandbox/159ee49257697c1d31904c383c87c004/default/test/einsum_trivia.ml.corrected index 35a36fa..335d797 100644 --- a/_build/default/test/einsum_trivia.ml +++ b/_build/.sandbox/159ee49257697c1d31904c383c87c004/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... - ┌─────────────────────────┐ - │[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 @@ 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 │ + │┌──────┬─────────────────────────────┐ │ + ││0 @ 0 │axis 3 │ │ + │├──────┼─────────────────────────────┤ │ + ││0 @ 1 │ -2.53e+37 3.07e-41 0.00 │ │ + ││axis 2│ -1.29e+37 2.91e+3 1.00 │ │ + ││ │ ... ... ... │ │ + ││ │ -1.29e+37 5.12e+3 4.00 │ │ + ││ │ -1.29e+37 0.00 5.00 │ │ + │├──────┼─────────────────────────────┤ │ + ││1 @ 1 │ -1.23e+37 1.00 2.06e+3 │ │ + ││axis 2│ 0.00 2.91e+3 2.06e+3 │ │ + ││ │ ... ... ... │ │ + ││ │ 4.68e+3 5.12e+3 2.06e+3 │ │ + ││ │ 0.00 1.00 2.06e+3 │ │ + │├──────┼─────────────────────────────┤ │ + ││2 @ 1 │ -1.23e+37 2.00 2.46e+3 │ │ + ││axis 2│ 2.14e+3 2.91e+3 2.46e+3 │ │ + ││ │ ... ... ... │ │ + ││ │ 6.82e+3 5.12e+3 2.46e+3 │ │ + ││ │ 2.14e+3 2.00 2.46e+3 │ │ + │├──────┼─────────────────────────────┤ │ + ││3 @ 1 │ -1.23e+37 3.00 2.86e+3 │ │ + ││axis 2│ 2.48e+3 2.91e+3 2.86e+3 │ │ + ││ │ ... ... ... │ │ + ││ │ 7.16e+3 5.12e+3 2.86e+3 │ │ + ││ │ 2.48e+3 3.00 2.86e+3 │ │ + │├──────┼─────────────────────────────┤ │ + ││4 @ 1 │ -1.23e+37 4.00 3.26e+3 │ │ + ││axis 2│ 2.82e+3 2.91e+3 3.26e+3 │ │ + ││ │ ... ... ... │ │ + ││ │ 7.50e+3 5.12e+3 3.26e+3 │ │ + ││ │ 2.82e+3 4.00 3.26e+3 │ │ + │└──────┴─────────────────────────────┘ │ + ├────────────────────────────────────────┤ + │┌──────┬───────────────────────────────┐│ + ││1 @ 0 │axis 3 ││ + │├──────┼───────────────────────────────┤│ + ││0 @ 1 │ -2.53e+37 2.89e+32 2.36e+21 ││ + ││axis 2│ -1.29e+37 2.89e+32 2.36e+21 ││ + ││ │ ... ... ... ││ + ││ │ -1.29e+37 2.89e+32 2.36e+21 ││ + ││ │ -1.29e+37 2.89e+32 2.36e+21 ││ + │├──────┼───────────────────────────────┤│ + ││1 @ 1 │ -1.23e+37 2.89e+32 2.36e+21 ││ + ││axis 2│ 1.00 2.89e+32 2.36e+21 ││ + ││ │ ... ... ... ││ + ││ │ 4.68e+3 2.89e+32 2.36e+21 ││ + ││ │ 1.00 2.89e+32 2.36e+21 ││ + │├──────┼───────────────────────────────┤│ + ││2 @ 1 │ -1.23e+37 2.89e+32 2.36e+21 ││ + ││axis 2│ 2.14e+3 2.89e+32 2.36e+21 ││ + ││ │ ... ... ... ││ + ││ │ 6.82e+3 2.89e+32 2.36e+21 ││ + ││ │ 2.14e+3 2.89e+32 2.36e+21 ││ + │├──────┼───────────────────────────────┤│ + ││3 @ 1 │ -1.23e+37 2.89e+32 2.36e+21 ││ + ││axis 2│ 2.48e+3 2.89e+32 2.36e+21 ││ + ││ │ ... ... ... ││ + ││ │ 7.16e+3 2.89e+32 2.36e+21 ││ + ││ │ 2.48e+3 2.89e+32 2.36e+21 ││ + │├──────┼───────────────────────────────┤│ + ││4 @ 1 │ -1.23e+37 2.89e+32 2.36e+21 ││ + ││axis 2│ 2.82e+3 2.89e+32 2.36e+21 ││ + ││ │ ... ... ... ││ + ││ │ 7.50e+3 2.89e+32 2.36e+21 ││ + ││ │ 2.82e+3 2.89e+32 2.36e+21 ││ + │└──────┴───────────────────────────────┘│ + ├────────────────────────────────────────┤ + │┌──────┬───────────────────────────────┐│ + ││2 @ 0 │axis 3 ││ + │├──────┼───────────────────────────────┤│ + ││0 @ 1 │ -2.53e+37 2.68e+23 1.63e+19 ││ + ││axis 2│ -1.29e+37 2.68e+23 1.63e+19 ││ + ││ │ ... ... ... ││ + ││ │ -1.29e+37 2.68e+23 1.63e+19 ││ + ││ │ -1.29e+37 2.68e+23 1.63e+19 ││ + │├──────┼───────────────────────────────┤│ + ││1 @ 1 │ -1.23e+37 2.68e+23 1.63e+19 ││ + ││axis 2│ 2.00 2.68e+23 1.63e+19 ││ + ││ │ ... ... ... ││ + ││ │ 4.68e+3 2.68e+23 1.63e+19 ││ + ││ │ 2.00 2.68e+23 1.63e+19 ││ + │├──────┼───────────────────────────────┤│ + ││2 @ 1 │ -1.23e+37 2.68e+23 1.63e+19 ││ + ││axis 2│ 2.15e+3 2.68e+23 1.63e+19 ││ + ││ │ ... ... ... ││ + ││ │ 6.83e+3 2.68e+23 1.63e+19 ││ + ││ │ 2.15e+3 2.68e+23 1.63e+19 ││ + │├──────┼───────────────────────────────┤│ + ││3 @ 1 │ -1.23e+37 2.68e+23 1.63e+19 ││ + ││axis 2│ 2.48e+3 2.68e+23 1.63e+19 ││ + ││ │ ... ... ... ││ + ││ │ 7.16e+3 2.68e+23 1.63e+19 ││ + ││ │ 2.48e+3 2.68e+23 1.63e+19 ││ + │├──────┼───────────────────────────────┤│ + ││4 @ 1 │ -1.23e+37 2.68e+23 1.63e+19 ││ + ││axis 2│ 2.82e+3 2.68e+23 1.63e+19 ││ + ││ │ ... ... ... ││ + ││ │ 7.50e+3 2.68e+23 1.63e+19 ││ + ││ │ 2.82e+3 2.68e+23 1.63e+19 ││ + │└──────┴───────────────────────────────┘│ + ├────────────────────────────────────────┤ + │┌──────┬──────────────────────────────┐ │ + ││3 @ 0 │axis 3 │ │ + │├──────┼──────────────────────────────┤ │ + ││0 @ 1 │ -2.53e+37 1.30e-8 7.52e+28 │ │ + ││axis 2│ -1.29e+37 2.91e+3 7.52e+28 │ │ + ││ │ ... ... ... │ │ + ││ │ -1.29e+37 5.12e+3 7.52e+28 │ │ + ││ │ -1.29e+37 1.30e-8 7.52e+28 │ │ + │├──────┼──────────────────────────────┤ │ + ││1 @ 1 │ -1.23e+37 1.00 7.52e+28 │ │ + ││axis 2│ 3.00 2.91e+3 7.52e+28 │ │ + ││ │ ... ... ... │ │ + ││ │ 4.68e+3 5.12e+3 7.52e+28 │ │ + ││ │ 3.00 1.00 7.52e+28 │ │ + │├──────┼──────────────────────────────┤ │ + ││2 @ 1 │ -1.23e+37 2.00 7.52e+28 │ │ + ││axis 2│ 2.15e+3 2.91e+3 7.52e+28 │ │ + ││ │ ... ... ... │ │ + ││ │ 6.83e+3 5.12e+3 7.52e+28 │ │ + ││ │ 2.15e+3 2.00 7.52e+28 │ │ + │├──────┼──────────────────────────────┤ │ + ││3 @ 1 │ -1.23e+37 3.00 7.52e+28 │ │ + ││axis 2│ 2.48e+3 2.91e+3 7.52e+28 │ │ + ││ │ ... ... ... │ │ + ││ │ 7.16e+3 5.12e+3 7.52e+28 │ │ + ││ │ 2.48e+3 3.00 7.52e+28 │ │ + │├──────┼──────────────────────────────┤ │ + ││4 @ 1 │ -1.23e+37 4.00 7.52e+28 │ │ + ││axis 2│ 2.82e+3 2.91e+3 7.52e+28 │ │ + ││ │ ... ... ... │ │ + ││ │ 7.50e+3 5.12e+3 7.52e+28 │ │ + ││ │ 2.82e+3 4.00 7.52e+28 │ │ + │└──────┴──────────────────────────────┘ │ + └────────────────────────────────────────┘ |}]; 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 0.00 0.00 ││ + ││ │ 1.00 2.89e+32 2.36e+21 ││ + ││ │ 2.00 2.68e+23 1.63e+19 ││ + ││ │ 3.00 1.30e-8 7.52e+28 ││ + │└──────┴──────────────────────────┘│ + └───────────────────────────────────┘ |}]; Tensor.print ~with_code:false ~with_grad:false `Default @@ tk; [%expect {| - ┌───────────────────────────┐ - │[7]: =>_tk shape 0:6,1:3 │ - │┌──────┬──────────────────┐│ - ││ │axis 1 ││ - │├──────┼──────────────────┤│ - ││axis 0│ 0.00 0.00 0.00 ││ - ││ │ 0.00 0.00 1.00 ││ - ││ │ ... ... ... ││ - ││ │ 0.00 0.00 4.00 ││ - ││ │ 0.00 0.00 5.00 ││ - │└──────┴──────────────────┘│ - └───────────────────────────┘ + ┌────────────────────────────────────┐ + │[7]: =>_tk shape 0:6,1:3 │ + │┌──────┬───────────────────────────┐│ + ││ │axis 1 ││ + │├──────┼───────────────────────────┤│ + ││axis 0│ -1.23e+37 3.07e-41 0.00 ││ + ││ │ 0.00 2.91e+3 1.00 ││ + ││ │ ... ... ... ││ + ││ │ 4.68e+3 5.12e+3 4.00 ││ + ││ │ 0.00 0.00 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 {| (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 #27} 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)))) @!Retrieving commandline, environment, or config file variable ocannl_prefer_backend_uniformity Found true, in the config file Retrieving commandline, environment, or config file variable ocannl_debug_log_to_stream_files Not found, using default false Retrieving commandline, environment, or config file variable ocannl_ll_ident_style Not found, using default heuristic Retrieving commandline, environment, or config file variable ocannl_cc_backend_optimization_level Not found, using default 3 Retrieving commandline, environment, or config file variable ocannl_cc_backend_compiler_command Not found, using default gcc Retrieving commandline, environment, or config file variable ocannl_cc_backend_verify_codesign Not found, using default false Retrieving commandline, environment, or config file variable ocannl_never_capture_stdout Not found, using default false Batch=59, step=60, lr=0.200000, batch loss=1.966667, epoch loss=1.966667 Batch=119, step=120, lr=0.199750, batch loss=0.000000, epoch loss=1.966667 Batch=179, step=180, lr=0.199500, batch loss=0.000000, epoch loss=1.966667 Batch=239, step=240, lr=0.199250, batch loss=0.000000, epoch loss=1.966667 Batch=299, step=300, lr=0.199000, batch loss=0.000000, epoch loss=1.966667 Batch=359, step=360, lr=0.198750, batch loss=0.000000, epoch loss=1.966667 Batch=419, step=420, lr=0.198500, batch loss=0.000000, epoch loss=1.966667 Batch=479, step=480, lr=0.198250, batch loss=0.000000, epoch loss=1.966667 Batch=539, step=540, lr=0.198000, batch loss=0.000000, epoch loss=1.966667 Batch=599, step=600, lr=0.197750, batch loss=0.000000, epoch loss=1.966667 Batch=659, step=660, lr=0.197500, batch loss=0.000000, epoch loss=1.966667 Batch=719, step=720, lr=0.197250, batch loss=0.000000, epoch loss=1.966667 Batch=779, step=780, lr=0.197000, batch loss=0.000000, epoch loss=1.966667 Batch=839, step=840, lr=0.196750, batch loss=0.000000, epoch loss=1.966667 Batch=899, step=900, lr=0.196500, batch loss=0.000000, epoch loss=1.966667 Batch=959, step=960, lr=0.196250, batch loss=0.000000, epoch loss=1.966667 Batch=1019, step=1020, lr=0.196000, batch loss=0.000000, epoch loss=1.966667 Batch=1079, step=1080, lr=0.195750, batch loss=0.000000, epoch loss=1.966667 Batch=1139, step=1140, lr=0.195500, batch loss=0.000000, epoch loss=1.966667 Batch=1199, step=1200, lr=0.195250, batch loss=0.000000, epoch loss=1.966667 Epoch=0, step=1200, lr=0.195250, epoch loss=1.966667 Batch=59, step=1260, lr=0.195000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=1320, lr=0.194750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=1380, lr=0.194500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=1440, lr=0.194250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=1500, lr=0.194000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=1560, lr=0.193750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=1620, lr=0.193500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=1680, lr=0.193250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=1740, lr=0.193000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=1800, lr=0.192750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=1860, lr=0.192500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=1920, lr=0.192250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=1980, lr=0.192000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=2040, lr=0.191750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=2100, lr=0.191500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=2160, lr=0.191250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=2220, lr=0.191000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=2280, lr=0.190750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=2340, lr=0.190500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=2400, lr=0.190250, batch loss=0.000000, epoch loss=0.000000 Epoch=1, step=2400, lr=0.190250, epoch loss=0.000000 Batch=59, step=2460, lr=0.190000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=2520, lr=0.189750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=2580, lr=0.189500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=2640, lr=0.189250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=2700, lr=0.189000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=2760, lr=0.188750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=2820, lr=0.188500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=2880, lr=0.188250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=2940, lr=0.188000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=3000, lr=0.187750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=3060, lr=0.187500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=3120, lr=0.187250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=3180, lr=0.187000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=3240, lr=0.186750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=3300, lr=0.186500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=3360, lr=0.186250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=3420, lr=0.186000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=3480, lr=0.185750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=3540, lr=0.185500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=3600, lr=0.185250, batch loss=0.000000, epoch loss=0.000000 Epoch=2, step=3600, lr=0.185250, epoch loss=0.000000 Batch=59, step=3660, lr=0.185000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=3720, lr=0.184750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=3780, lr=0.184500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=3840, lr=0.184250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=3900, lr=0.184000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=3960, lr=0.183750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=4020, lr=0.183500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=4080, lr=0.183250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=4140, lr=0.183000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=4200, lr=0.182750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=4260, lr=0.182500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=4320, lr=0.182250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=4380, lr=0.182000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=4440, lr=0.181750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=4500, lr=0.181500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=4560, lr=0.181250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=4620, lr=0.181000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=4680, lr=0.180750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=4740, lr=0.180500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=4800, lr=0.180250, batch loss=0.000000, epoch loss=0.000000 Epoch=3, step=4800, lr=0.180250, epoch loss=0.000000 Batch=59, step=4860, lr=0.180000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=4920, lr=0.179750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=4980, lr=0.179500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=5040, lr=0.179250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=5100, lr=0.179000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=5160, lr=0.178750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=5220, lr=0.178500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=5280, lr=0.178250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=5340, lr=0.178000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=5400, lr=0.177750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=5460, lr=0.177500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=5520, lr=0.177250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=5580, lr=0.177000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=5640, lr=0.176750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=5700, lr=0.176500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=5760, lr=0.176250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=5820, lr=0.176000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=5880, lr=0.175750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=5940, lr=0.175500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=6000, lr=0.175250, batch loss=0.000000, epoch loss=0.000000 Epoch=4, step=6000, lr=0.175250, epoch loss=0.000000 Batch=59, step=6060, lr=0.175000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=6120, lr=0.174750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=6180, lr=0.174500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=6240, lr=0.174250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=6300, lr=0.174000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=6360, lr=0.173750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=6420, lr=0.173500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=6480, lr=0.173250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=6540, lr=0.173000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=6600, lr=0.172750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=6660, lr=0.172500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=6720, lr=0.172250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=6780, lr=0.172000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=6840, lr=0.171750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=6900, lr=0.171500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=6960, lr=0.171250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=7020, lr=0.171000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=7080, lr=0.170750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=7140, lr=0.170500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=7200, lr=0.170250, batch loss=0.000000, epoch loss=0.000000 Epoch=5, step=7200, lr=0.170250, epoch loss=0.000000 Batch=59, step=7260, lr=0.170000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=7320, lr=0.169750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=7380, lr=0.169500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=7440, lr=0.169250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=7500, lr=0.169000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=7560, lr=0.168750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=7620, lr=0.168500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=7680, lr=0.168250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=7740, lr=0.168000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=7800, lr=0.167750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=7860, lr=0.167500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=7920, lr=0.167250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=7980, lr=0.167000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=8040, lr=0.166750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=8100, lr=0.166500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=8160, lr=0.166250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=8220, lr=0.166000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=8280, lr=0.165750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=8340, lr=0.165500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=8400, lr=0.165250, batch loss=0.000000, epoch loss=0.000000 Epoch=6, step=8400, lr=0.165250, epoch loss=0.000000 Batch=59, step=8460, lr=0.165000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=8520, lr=0.164750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=8580, lr=0.164500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=8640, lr=0.164250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=8700, lr=0.164000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=8760, lr=0.163750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=8820, lr=0.163500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=8880, lr=0.163250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=8940, lr=0.163000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=9000, lr=0.162750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=9060, lr=0.162500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=9120, lr=0.162250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=9180, lr=0.162000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=9240, lr=0.161750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=9300, lr=0.161500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=9360, lr=0.161250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=9420, lr=0.161000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=9480, lr=0.160750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=9540, lr=0.160500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=9600, lr=0.160250, batch loss=0.000000, epoch loss=0.000000 Epoch=7, step=9600, lr=0.160250, epoch loss=0.000000 Batch=59, step=9660, lr=0.160000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=9720, lr=0.159750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=9780, lr=0.159500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=9840, lr=0.159250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=9900, lr=0.159000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=9960, lr=0.158750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=10020, lr=0.158500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=10080, lr=0.158250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=10140, lr=0.158000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=10200, lr=0.157750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=10260, lr=0.157500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=10320, lr=0.157250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=10380, lr=0.157000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=10440, lr=0.156750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=10500, lr=0.156500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=10560, lr=0.156250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=10620, lr=0.156000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=10680, lr=0.155750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=10740, lr=0.155500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=10800, lr=0.155250, batch loss=0.000000, epoch loss=0.000000 Epoch=8, step=10800, lr=0.155250, epoch loss=0.000000 Batch=59, step=10860, lr=0.155000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=10920, lr=0.154750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=10980, lr=0.154500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=11040, lr=0.154250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=11100, lr=0.154000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=11160, lr=0.153750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=11220, lr=0.153500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=11280, lr=0.153250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=11340, lr=0.153000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=11400, lr=0.152750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=11460, lr=0.152500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=11520, lr=0.152250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=11580, lr=0.152000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=11640, lr=0.151750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=11700, lr=0.151500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=11760, lr=0.151250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=11820, lr=0.151000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=11880, lr=0.150750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=11940, lr=0.150500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=12000, lr=0.150250, batch loss=0.000000, epoch loss=0.000000 Epoch=9, step=12000, lr=0.150250, epoch loss=0.000000 Batch=59, step=12060, lr=0.150000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=12120, lr=0.149750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=12180, lr=0.149500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=12240, lr=0.149250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=12300, lr=0.149000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=12360, lr=0.148750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=12420, lr=0.148500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=12480, lr=0.148250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=12540, lr=0.148000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=12600, lr=0.147750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=12660, lr=0.147500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=12720, lr=0.147250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=12780, lr=0.147000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=12840, lr=0.146750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=12900, lr=0.146500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=12960, lr=0.146250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=13020, lr=0.146000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=13080, lr=0.145750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=13140, lr=0.145500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=13200, lr=0.145250, batch loss=0.000000, epoch loss=0.000000 Epoch=10, step=13200, lr=0.145250, epoch loss=0.000000 Batch=59, step=13260, lr=0.145000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=13320, lr=0.144750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=13380, lr=0.144500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=13440, lr=0.144250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=13500, lr=0.144000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=13560, lr=0.143750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=13620, lr=0.143500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=13680, lr=0.143250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=13740, lr=0.143000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=13800, lr=0.142750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=13860, lr=0.142500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=13920, lr=0.142250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=13980, lr=0.142000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=14040, lr=0.141750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=14100, lr=0.141500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=14160, lr=0.141250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=14220, lr=0.141000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=14280, lr=0.140750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=14340, lr=0.140500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=14400, lr=0.140250, batch loss=0.000000, epoch loss=0.000000 Epoch=11, step=14400, lr=0.140250, epoch loss=0.000000 Batch=59, step=14460, lr=0.140000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=14520, lr=0.139750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=14580, lr=0.139500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=14640, lr=0.139250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=14700, lr=0.139000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=14760, lr=0.138750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=14820, lr=0.138500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=14880, lr=0.138250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=14940, lr=0.138000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=15000, lr=0.137750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=15060, lr=0.137500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=15120, lr=0.137250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=15180, lr=0.137000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=15240, lr=0.136750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=15300, lr=0.136500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=15360, lr=0.136250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=15420, lr=0.136000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=15480, lr=0.135750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=15540, lr=0.135500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=15600, lr=0.135250, batch loss=0.000000, epoch loss=0.000000 Epoch=12, step=15600, lr=0.135250, epoch loss=0.000000 Batch=59, step=15660, lr=0.135000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=15720, lr=0.134750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=15780, lr=0.134500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=15840, lr=0.134250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=15900, lr=0.134000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=15960, lr=0.133750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=16020, lr=0.133500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=16080, lr=0.133250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=16140, lr=0.133000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=16200, lr=0.132750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=16260, lr=0.132500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=16320, lr=0.132250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=16380, lr=0.132000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=16440, lr=0.131750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=16500, lr=0.131500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=16560, lr=0.131250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=16620, lr=0.131000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=16680, lr=0.130750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=16740, lr=0.130500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=16800, lr=0.130250, batch loss=0.000000, epoch loss=0.000000 Epoch=13, step=16800, lr=0.130250, epoch loss=0.000000 Batch=59, step=16860, lr=0.130000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=16920, lr=0.129750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=16980, lr=0.129500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=17040, lr=0.129250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=17100, lr=0.129000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=17160, lr=0.128750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=17220, lr=0.128500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=17280, lr=0.128250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=17340, lr=0.128000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=17400, lr=0.127750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=17460, lr=0.127500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=17520, lr=0.127250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=17580, lr=0.127000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=17640, lr=0.126750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=17700, lr=0.126500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=17760, lr=0.126250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=17820, lr=0.126000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=17880, lr=0.125750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=17940, lr=0.125500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=18000, lr=0.125250, batch loss=0.000000, epoch loss=0.000000 Epoch=14, step=18000, lr=0.125250, epoch loss=0.000000 Batch=59, step=18060, lr=0.125000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=18120, lr=0.124750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=18180, lr=0.124500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=18240, lr=0.124250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=18300, lr=0.124000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=18360, lr=0.123750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=18420, lr=0.123500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=18480, lr=0.123250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=18540, lr=0.123000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=18600, lr=0.122750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=18660, lr=0.122500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=18720, lr=0.122250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=18780, lr=0.122000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=18840, lr=0.121750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=18900, lr=0.121500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=18960, lr=0.121250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=19020, lr=0.121000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=19080, lr=0.120750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=19140, lr=0.120500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=19200, lr=0.120250, batch loss=0.000000, epoch loss=0.000000 Epoch=15, step=19200, lr=0.120250, epoch loss=0.000000 Batch=59, step=19260, lr=0.120000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=19320, lr=0.119750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=19380, lr=0.119500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=19440, lr=0.119250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=19500, lr=0.119000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=19560, lr=0.118750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=19620, lr=0.118500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=19680, lr=0.118250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=19740, lr=0.118000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=19800, lr=0.117750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=19860, lr=0.117500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=19920, lr=0.117250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=19980, lr=0.117000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=20040, lr=0.116750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=20100, lr=0.116500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=20160, lr=0.116250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=20220, lr=0.116000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=20280, lr=0.115750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=20340, lr=0.115500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=20400, lr=0.115250, batch loss=0.000000, epoch loss=0.000000 Epoch=16, step=20400, lr=0.115250, epoch loss=0.000000 Batch=59, step=20460, lr=0.115000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=20520, lr=0.114750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=20580, lr=0.114500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=20640, lr=0.114250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=20700, lr=0.114000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=20760, lr=0.113750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=20820, lr=0.113500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=20880, lr=0.113250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=20940, lr=0.113000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=21000, lr=0.112750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=21060, lr=0.112500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=21120, lr=0.112250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=21180, lr=0.112000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=21240, lr=0.111750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=21300, lr=0.111500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=21360, lr=0.111250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=21420, lr=0.111000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=21480, lr=0.110750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=21540, lr=0.110500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=21600, lr=0.110250, batch loss=0.000000, epoch loss=0.000000 Epoch=17, step=21600, lr=0.110250, epoch loss=0.000000 Batch=59, step=21660, lr=0.110000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=21720, lr=0.109750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=21780, lr=0.109500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=21840, lr=0.109250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=21900, lr=0.109000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=21960, lr=0.108750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=22020, lr=0.108500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=22080, lr=0.108250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=22140, lr=0.108000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=22200, lr=0.107750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=22260, lr=0.107500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=22320, lr=0.107250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=22380, lr=0.107000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=22440, lr=0.106750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=22500, lr=0.106500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=22560, lr=0.106250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=22620, lr=0.106000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=22680, lr=0.105750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=22740, lr=0.105500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=22800, lr=0.105250, batch loss=0.000000, epoch loss=0.000000 Epoch=18, step=22800, lr=0.105250, epoch loss=0.000000 Batch=59, step=22860, lr=0.105000, batch loss=0.000000, epoch loss=0.000000 Batch=119, step=22920, lr=0.104750, batch loss=0.000000, epoch loss=0.000000 Batch=179, step=22980, lr=0.104500, batch loss=0.000000, epoch loss=0.000000 Batch=239, step=23040, lr=0.104250, batch loss=0.000000, epoch loss=0.000000 Batch=299, step=23100, lr=0.104000, batch loss=0.000000, epoch loss=0.000000 Batch=359, step=23160, lr=0.103750, batch loss=0.000000, epoch loss=0.000000 Batch=419, step=23220, lr=0.103500, batch loss=0.000000, epoch loss=0.000000 Batch=479, step=23280, lr=0.103250, batch loss=0.000000, epoch loss=0.000000 Batch=539, step=23340, lr=0.103000, batch loss=0.000000, epoch loss=0.000000 Batch=599, step=23400, lr=0.102750, batch loss=0.000000, epoch loss=0.000000 Batch=659, step=23460, lr=0.102500, batch loss=0.000000, epoch loss=0.000000 Batch=719, step=23520, lr=0.102250, batch loss=0.000000, epoch loss=0.000000 Batch=779, step=23580, lr=0.102000, batch loss=0.000000, epoch loss=0.000000 Batch=839, step=23640, lr=0.101750, batch loss=0.000000, epoch loss=0.000000 Batch=899, step=23700, lr=0.101500, batch loss=0.000000, epoch loss=0.000000 Batch=959, step=23760, lr=0.101250, batch loss=0.000000, epoch loss=0.000000 Batch=1019, step=23820, lr=0.101000, batch loss=0.000000, epoch loss=0.000000 Batch=1079, step=23880, lr=0.100750, batch loss=0.000000, epoch loss=0.000000 Batch=1139, step=23940, lr=0.100500, batch loss=0.000000, epoch loss=0.000000 Batch=1199, step=24000, lr=0.100250, batch loss=0.000000, epoch loss=0.000000 Epoch=19, step=24000, lr=0.100250, epoch loss=0.000000 Half-moons scatterplot and decision boundary: ┌────────────────────────────────────────────────────────────────────────────────────────────────────┐ │...................................#................................................................│ │.......................#.#################.#........................................................│ │......................##########################....................................................│ │.................#################################..................................................│ │..............######################################................................................│ │.............#########################################..............................................│ │...........################.............#################...........................................│ │..........##############...................###############..........................................│ │........##############..........................###########.........................................│ │.......#############............................##.##########.......................................│ │......###########..................................############.....................................│ │.....###########....................................###########.....................................│ │....##########.....................%.................###########....................................│ │...#########.#...................%%%%.................#########..............................%%%.%%.│ │...##########..................%%%.%%...................########.#...........................%%.%%%.│ │..#######......................%%%%%%%..................##########...........................%%%%%%%│ │.#########....................%%%%%%%%....................########...........................%%%%%%.│ │.########......................%%%%%%%.....................########..........................%%%%%%.│ │#########.....................%%.%%%%%......................########........................%%%%%%%.│ │.#######.......................%%%%%%%%....................#########........................%%%%%%%.│ │########.......................%%%%%%%%.....................########.......................%%%%%%%%.│ │#######.........................%%%%%%%......................#######.......................%%%%%%%%.│ │#######.........................%%%%%%%%......................#######.....................%%%%%%%%..│ │#######.........................%%%%%%%%.....................#######.....................%%%%%%%%%..│ │######............................%%%%%%%%...................#######....................%%%%%%%%%...│ │######...........................%%%%%%%%....................########..................%%%%%%%%%%...│ │######............................%%%%%%%%%..................#######..................%.%%%%%%%%....│ │###.#.............................%.%%%%%%%%%.................#.#.##..................%%.%%%%%%%....│ │....................................%%%%%%%%%.......................................%%%%%%%%%%%.....│ │......................................%%%%%%%.%.....................................%%%%%%%%%%%.....│ │.....................................%%%%%%%%%%%.................................%%%%%%%%%%%%.......│ │......................................%%.%%%%%%%%...............................%%%%%%%%%%%.........│ │........................................%%%%%%%%%%%%........................%..%%%%%%%%%%%%.........│ │........................................%%%%%%%%%%%%%%.%....................%%%%%%%%%%%%%...........│ │..........................................%%%%%%%%%%%%%%%%%..............%%%%%%%%%%%%%%%............│ │............................................%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%...............│ │................................................%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%.%..%..............│ │................................................%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%.%..................│ │..................................................%%%%%%%%%%%%%%%%%%%%%%%%%%%%.%....................│ │.....................................................%%.%..%%%%%%%%%%%%%%%..........................│ └────────────────────────────────────────────────────────────────────────────────────────────────────┘ "/usr/bin/env" "bash" "-c" "opam exec -- dune build @install @check @runtest && rm -rf _build" failed with exit status 1 2025-07-05 17:58.13: Job failed: Failed: Build failed