am'¶

In [ ]:
//// test

open testing
In [ ]:
open rust
open rust_operators

am'¶

length¶

In [ ]:
inl length forall dim {int} el. (a : a dim el) : dim =
    a |> length

index¶

In [ ]:
inl index forall dim {int} el. (i : dim) (a : a dim el) : el =
    backend_switch {
        Fsharp = fun () => index a i
        Python = fun () => $'!a[!i]' : el
    }

index_base¶

In [ ]:
inl index_base forall el. (i : int) (ar : array_base el) : el =
    ar |> a |> index i

base¶

In [ ]:
inl base forall dim {int} el. ((a a) : a dim el) : array_base el =
    a

base'¶

In [ ]:
inl base' forall el. ((a a) : a int el) : array_base el =
    a

generic_equable¶

In [ ]:
inl generic_equable x1 x2 =
    if length x1 <>.. length x2
    then false
    else
        let rec loop i =
            if i < length x1
            then index i x1 = index i x2 && loop (i + 1)
            else true
        loop 0

equable¶

In [ ]:
instance equable a dim {number; int} t = generic_equable

append¶

In [ ]:
instance append a dim {int; number} t = am.append
In [ ]:
//// test
///! fsharp
///! cuda
///! rust
///! typescript
///! python

a' ;[ -1i64; 0 ] ++ a' ;[ 1; 2 ]
|> _assert_eq (a' ;[ -1; 0; 1; 2 ])

.py output (Cuda):
Traceback (most recent call last):
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/68b9bb09c502ee0e564185d4d3fcabb90047190021dd0f7720e2e9dc1b639e8e/main.py", line 164, in <module>
    if __name__ == '__main__': result = main(); None if result is None else print(result)
                                        ^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/68b9bb09c502ee0e564185d4d3fcabb90047190021dd0f7720e2e9dc1b639e8e/main.py", line 160, in main
    r = main_body()
        ^^^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/68b9bb09c502ee0e564185d4d3fcabb90047190021dd0f7720e2e9dc1b639e8e/main.py", line 157, in main_body
    return method0()
           ^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/68b9bb09c502ee0e564185d4d3fcabb90047190021dd0f7720e2e9dc1b639e8e/main.py", line 96, in method0
    v0 = cp.array([-1, 0],dtype=cp.int64)
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/site-packages/cupy/_creation/from_data.py", line 53, in array
    return _core.array(obj, dtype, copy, order, subok, ndmin, blocking)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "cupy/_core/core.pyx", line 2408, in cupy._core.core.array
  File "cupy/_core/core.pyx", line 2435, in cupy._core.core.array
  File "cupy/_core/core.pyx", line 2578, in cupy._core.core._array_default
  File "cupy/_core/core.pyx", line 137, in cupy._core.core.ndarray.__new__
  File "cupy/_core/core.pyx", line 225, in cupy._core.core._ndarray_base._init
  File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1444, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/device.pyx", line 40, in cupy.cuda.device.get_device_id
  File "cupy_backends/cuda/api/runtime.pyx", line 202, in cupy_backends.cuda.api.runtime.getDevice
  File "cupy_backends/cuda/api/runtime.pyx", line 146, in cupy_backends.cuda.api.runtime.check_status
cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version

.rs output:
__assert_eq / actual: Array(MutCell([-1, 0, 1, 2])) / expected: Array(MutCell([-1, 0, 1, 2]))

.ts output:
__assert_eq / actual: -1,0,1,2 / expected: -1,0,1,2

.py output:
__assert_eq / actual: [-1, 0, 1, 2] / expected: array('q', [-1, 0, 1, 2])


.fsx output:
__assert_eq / actual: [|-1L; 0L; 1L; 2L|] / expected: [|-1L; 0L; 1L; 2L|]

map_base¶

In [ ]:
inl map_base forall t u. (fn : t -> u) (x : array_base t) : array_base u =
    a x
    |> am.map fn
    |> fun (a x : _ int _) => x

collect¶

In [ ]:
inl collect forall t r. (fn : t -> a int r) (items : a int t) : a int r =
    items
    |> am.map fn
    |> am.fold (++) (a ;[])

init¶

In [ ]:
inl init n : array_base _ =
    am.init n id
    |> fun (a x : _ int _) => x

choose¶

In [ ]:
inl choose f l =
    (l, [])
    ||> am.foldBack fun x acc =>
        match f x with
        | Some y => y :: acc
        | None => acc
    |> listm.toArray
In [ ]:
//// test
///! fsharp
///! cuda
////! rust // v0.get_mut()[v2.get().clone() as usize] = match v1.get().clone().as_ref() { ^ expected `i32`, found `usize`
///! typescript
///! python

10
|> init
|> fun x => a x : _ int _
|> choose (fun x => if x % 2 = 0 then Some x else None)
|> _assert_eq (a' ;[ 0; 2; 4; 6; 8 ])

.py output (Cuda):
Traceback (most recent call last):
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/9d4ad739531a383d55562d2d54e83a808122db5c2290b3d9af7adfba08f40e71/main.py", line 245, in <module>
    if __name__ == '__main__': result = main(); None if result is None else print(result)
                                        ^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/9d4ad739531a383d55562d2d54e83a808122db5c2290b3d9af7adfba08f40e71/main.py", line 241, in main
    r = main_body()
        ^^^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/9d4ad739531a383d55562d2d54e83a808122db5c2290b3d9af7adfba08f40e71/main.py", line 238, in main_body
    return method0()
           ^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/9d4ad739531a383d55562d2d54e83a808122db5c2290b3d9af7adfba08f40e71/main.py", line 155, in method0
    v0 = cp.empty(10,dtype=cp.int32)
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/site-packages/cupy/_creation/basic.py", line 31, in empty
    return cupy.ndarray(shape, dtype, order=order)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "cupy/_core/core.pyx", line 137, in cupy._core.core.ndarray.__new__
  File "cupy/_core/core.pyx", line 225, in cupy._core.core._ndarray_base._init
  File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1444, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/device.pyx", line 40, in cupy.cuda.device.get_device_id
  File "cupy_backends/cuda/api/runtime.pyx", line 202, in cupy_backends.cuda.api.runtime.getDevice
  File "cupy_backends/cuda/api/runtime.pyx", line 146, in cupy_backends.cuda.api.runtime.check_status
cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version

.ts output:
__assert_eq / actual: 0,2,4,6,8 / expected: 0,2,4,6,8

.py output:
__assert_eq / actual: [0, 2, 4, 6, 8] / expected: array('l', [0, 2, 4, 6, 8])


.fsx output:
__assert_eq / actual: [|0; 2; 4; 6; 8|] / expected: [|0; 2; 4; 6; 8|]

sum¶

In [ ]:
inl sum a =
    a |> am.fold (+) 0
In [ ]:
//// test
///! fsharp
///! cuda
///! rust
///! typescript
///! python

10
|> init
|> fun x => a x : _ int _
|> sum
|> _assert_eq 45

.py output (Cuda):
Traceback (most recent call last):
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/d2ff92bc98ea55b61391a67d3568f034e4f65522e461ab8b56245677d24c8308/main.py", line 129, in <module>
    if __name__ == '__main__': result = main(); None if result is None else print(result)
                                        ^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/d2ff92bc98ea55b61391a67d3568f034e4f65522e461ab8b56245677d24c8308/main.py", line 125, in main
    r = main_body()
        ^^^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/d2ff92bc98ea55b61391a67d3568f034e4f65522e461ab8b56245677d24c8308/main.py", line 122, in main_body
    return method0()
           ^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/d2ff92bc98ea55b61391a67d3568f034e4f65522e461ab8b56245677d24c8308/main.py", line 77, in method0
    v0 = cp.empty(10,dtype=cp.int32)
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/site-packages/cupy/_creation/basic.py", line 31, in empty
    return cupy.ndarray(shape, dtype, order=order)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "cupy/_core/core.pyx", line 137, in cupy._core.core.ndarray.__new__
  File "cupy/_core/core.pyx", line 225, in cupy._core.core._ndarray_base._init
  File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1444, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/device.pyx", line 40, in cupy.cuda.device.get_device_id
  File "cupy_backends/cuda/api/runtime.pyx", line 202, in cupy_backends.cuda.api.runtime.getDevice
  File "cupy_backends/cuda/api/runtime.pyx", line 146, in cupy_backends.cuda.api.runtime.check_status
cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version

.rs output:
__assert_eq / actual: 45 / expected: 45

.ts output:
__assert_eq / actual: 45 / expected: 45

.py output:
__assert_eq / actual: 45 / expected: 45


.fsx output:
__assert_eq / actual: 45 / expected: 45

init_series¶

In [ ]:
inl init_series start end inc : array_base _ =
    inl total = conv ((end - start) / inc) + 1
    am.init total (conv >> (*) inc >> (+) start)
    |> fun (a x : _ int _) => x
In [ ]:
//// test
///! fsharp
///! cuda
///! rust
///! typescript
///! python

init_series 0 4 2
|> _assert_eq' ;[ 0i32; 2; 4 ]

.py output (Cuda):
Traceback (most recent call last):
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/cc6bc9f685a23ff0387e2d5adeefc74ebf5530960aa10194948f6cbeed681926/main.py", line 106, in <module>
    if __name__ == '__main__': result = main(); None if result is None else print(result)
                                        ^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/cc6bc9f685a23ff0387e2d5adeefc74ebf5530960aa10194948f6cbeed681926/main.py", line 102, in main
    r = main_body()
        ^^^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/cc6bc9f685a23ff0387e2d5adeefc74ebf5530960aa10194948f6cbeed681926/main.py", line 99, in main_body
    return method0()
           ^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/cc6bc9f685a23ff0387e2d5adeefc74ebf5530960aa10194948f6cbeed681926/main.py", line 67, in method0
    v0 = cp.empty(3,dtype=cp.int32)
         ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/site-packages/cupy/_creation/basic.py", line 31, in empty
    return cupy.ndarray(shape, dtype, order=order)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "cupy/_core/core.pyx", line 137, in cupy._core.core.ndarray.__new__
  File "cupy/_core/core.pyx", line 225, in cupy._core.core._ndarray_base._init
  File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1444, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/device.pyx", line 40, in cupy.cuda.device.get_device_id
  File "cupy_backends/cuda/api/runtime.pyx", line 202, in cupy_backends.cuda.api.runtime.getDevice
  File "cupy_backends/cuda/api/runtime.pyx", line 146, in cupy_backends.cuda.api.runtime.check_status
cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version

.rs output:
__assert_eq' / actual: Array(MutCell([0, 2, 4])) / expected: Array(MutCell([0, 2, 4]))

.ts output:
__assert_eq' / actual: 0,2,4 / expected: 0,2,4

.py output:
__assert_eq' / actual: [0, 2, 4] / expected: array('l', [0, 2, 4])


.fsx output:
__assert_eq' / actual: [|0; 2; 4|] / expected: [|0; 2; 4|]

head¶

In [ ]:
inl head (ar : a _ _) =
    if var_is ar || length ar > 0
    then ar |> index 0
    else error_type "The length of the array should be greater than 0."

last¶

In [ ]:
inl last (ar : a _ _) =
    inl len = length ar
    if var_is ar || len > 0
    then ar |> index (len - 1)
    else error_type "The length of the array should be greater than 0."
In [ ]:
//// test
///! fsharp
///! cuda
///! rust
///! typescript
///! python

10
|> init
|> fun x => a x : _ int _
|> last
|> _assert_eq 9

.py output (Cuda):
Traceback (most recent call last):
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/ed1d249488022df51e5f876496d2021ed16162ab1f9734f72e17cda34328ce10/main.py", line 108, in <module>
    if __name__ == '__main__': result = main(); None if result is None else print(result)
                                        ^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/ed1d249488022df51e5f876496d2021ed16162ab1f9734f72e17cda34328ce10/main.py", line 104, in main
    r = main_body()
        ^^^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/ed1d249488022df51e5f876496d2021ed16162ab1f9734f72e17cda34328ce10/main.py", line 101, in main_body
    return method0()
           ^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/ed1d249488022df51e5f876496d2021ed16162ab1f9734f72e17cda34328ce10/main.py", line 67, in method0
    v0 = cp.empty(10,dtype=cp.int32)
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/site-packages/cupy/_creation/basic.py", line 31, in empty
    return cupy.ndarray(shape, dtype, order=order)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "cupy/_core/core.pyx", line 137, in cupy._core.core.ndarray.__new__
  File "cupy/_core/core.pyx", line 225, in cupy._core.core._ndarray_base._init
  File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1444, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/device.pyx", line 40, in cupy.cuda.device.get_device_id
  File "cupy_backends/cuda/api/runtime.pyx", line 202, in cupy_backends.cuda.api.runtime.getDevice
  File "cupy_backends/cuda/api/runtime.pyx", line 146, in cupy_backends.cuda.api.runtime.check_status
cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version

.rs output:
__assert_eq / actual: 9 / expected: 9

.ts output:
__assert_eq / actual: 9 / expected: 9

.py output:
__assert_eq / actual: 9 / expected: 9


.fsx output:
__assert_eq / actual: 9 / expected: 9

try_pick¶

In [ ]:
inl try_pick forall t u. (fn : t -> option u) (array : a _ t) : option u =
    (array, None)
    ||> am.foldBack fun x acc =>
        match acc with
        | Some _ => acc
        | None => x |> fn
In [ ]:
//// test
///! fsharp
///! cuda
////! rust // match &v23 { Spiral_builder::US0::US0_0(x) => x.clone(), _ => unreachable!(), } == 5_i32
///! typescript
///! python

10
|> init
|> fun x => a x : _ int _
|> try_pick (fun x => if x = 5i32 then Some x else None)
|> _assert_eq (Some 5i32)

.py output (Cuda):
Traceback (most recent call last):
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/f574f6aa78dd2d474cc9a15e9816e77781ab6f25f1c539d19fe6f862f7ac5433/main.py", line 162, in <module>
    if __name__ == '__main__': result = main(); None if result is None else print(result)
                                        ^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/f574f6aa78dd2d474cc9a15e9816e77781ab6f25f1c539d19fe6f862f7ac5433/main.py", line 158, in main
    r = main_body()
        ^^^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/f574f6aa78dd2d474cc9a15e9816e77781ab6f25f1c539d19fe6f862f7ac5433/main.py", line 155, in main_body
    return method0()
           ^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/f574f6aa78dd2d474cc9a15e9816e77781ab6f25f1c539d19fe6f862f7ac5433/main.py", line 83, in method0
    v0 = cp.empty(10,dtype=cp.int32)
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/site-packages/cupy/_creation/basic.py", line 31, in empty
    return cupy.ndarray(shape, dtype, order=order)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "cupy/_core/core.pyx", line 137, in cupy._core.core.ndarray.__new__
  File "cupy/_core/core.pyx", line 225, in cupy._core.core._ndarray_base._init
  File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1444, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/device.pyx", line 40, in cupy.cuda.device.get_device_id
  File "cupy_backends/cuda/api/runtime.pyx", line 202, in cupy_backends.cuda.api.runtime.getDevice
  File "cupy_backends/cuda/api/runtime.pyx", line 146, in cupy_backends.cuda.api.runtime.check_status
cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version

.ts output:
__assert_eq / actual: US0_0 5 / expected: US0_0 5

.py output:
__assert_eq / actual: US0_0 5 / expected: US0_0 5


.fsx output:
__assert_eq / actual: US0_0 5 / expected: US0_0 5

indexed¶

In [ ]:
inl indexed forall t u {number}. (ar : array_base t) : array_base (u * t) =
    ((0, a ;[]), (a ar : _ int _))
    ||> am.fold fun (i, acc) x =>
        i + 1, acc ++ a ;[i, x]
    |> snd
    |> fun (a x : _ int _) => x
In [ ]:
//// test
///! fsharp
////! cuda // Only stack allocated primitive types (i8,i16,i32,i64 and u8,u16,u32,u64 and f32,f64 and bool) are allowed in CuPy arrays.
///! rust
///! typescript
///! python

am.init 3i32 ((*) 2)
|> fun (a x : _ int _) => x
|> indexed
|> _assert_eq' ;[0i32, 0; 1, 2; 2, 4]

.rs output:
__assert_eq' / actual: Array(MutCell([(0, 0), (1, 2), (2, 4)])) / expected: Array(MutCell([(0, 0), (1, 2), (2, 4)]))

.ts output:
__assert_eq' / actual: 0,0,1,2,2,4 / expected: 0,0,1,2,2,4

.py output:
__assert_eq' / actual: [(0, 0), (1, 2), (2, 4)] / expected: [(0, 0), (1, 2), (2, 4)]


.fsx output:
__assert_eq' / actual: [|struct (0, 0); struct (1, 2); struct (2, 4)|] / expected: [|struct (0, 0); struct (1, 2); struct (2, 4)|]

slice¶

In [ ]:
inl slice forall dim {int; number} el. from nearTo s : a dim el =
    am.slice { from nearTo } s
In [ ]:
//// test
///! fsharp
///! cuda
///! rust
///! typescript
///! python

inl x : _ i32 _ = a ;[ 1i32; 2; 3 ]
x |> slice 0 0 |> _assert_eq (a ;[])
x |> slice 0 1 |> _assert_eq (a ;[ 1 ])
x |> slice 1 1 |> _assert_eq (a ;[])
x |> slice 1 2 |> _assert_eq (a ;[ 2 ])
x |> slice 2 2 |> _assert_eq (a ;[])
x |> slice 0 2 |> _assert_eq (a ;[ 1; 2 ])

.py output (Cuda):
Traceback (most recent call last):
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/a8adc9f979fb0d14587fb26fafe8b2f1cddc6a9cfe0d328177faf8a154a65c7b/main.py", line 372, in <module>
    if __name__ == '__main__': result = main(); None if result is None else print(result)
                                        ^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/a8adc9f979fb0d14587fb26fafe8b2f1cddc6a9cfe0d328177faf8a154a65c7b/main.py", line 368, in main
    r = main_body()
        ^^^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/a8adc9f979fb0d14587fb26fafe8b2f1cddc6a9cfe0d328177faf8a154a65c7b/main.py", line 365, in main_body
    return method0()
           ^^^^^^^^^
  File "/home/runner/work/polyglot/polyglot/target/spiral_Eval/packages/a8adc9f979fb0d14587fb26fafe8b2f1cddc6a9cfe0d328177faf8a154a65c7b/main.py", line 108, in method0
    v0 = cp.array([1, 2, 3],dtype=cp.int32)
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/site-packages/cupy/_creation/from_data.py", line 53, in array
    return _core.array(obj, dtype, copy, order, subok, ndmin, blocking)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "cupy/_core/core.pyx", line 2408, in cupy._core.core.array
  File "cupy/_core/core.pyx", line 2435, in cupy._core.core.array
  File "cupy/_core/core.pyx", line 2578, in cupy._core.core._array_default
  File "cupy/_core/core.pyx", line 137, in cupy._core.core.ndarray.__new__
  File "cupy/_core/core.pyx", line 225, in cupy._core.core._ndarray_base._init
  File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1444, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/device.pyx", line 40, in cupy.cuda.device.get_device_id
  File "cupy_backends/cuda/api/runtime.pyx", line 202, in cupy_backends.cuda.api.runtime.getDevice
  File "cupy_backends/cuda/api/runtime.pyx", line 146, in cupy_backends.cuda.api.runtime.check_status
cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version


.rs output:
__assert_eq / actual: Array(MutCell([])) / expected: Array(MutCell([]))
__assert_eq / actual: Array(MutCell([1])) / expected: Array(MutCell([1]))
__assert_eq / actual: Array(MutCell([])) / expected: Array(MutCell([]))
__assert_eq / actual: Array(MutCell([2])) / expected: Array(MutCell([2]))
__assert_eq / actual: Array(MutCell([])) / expected: Array(MutCell([]))
__assert_eq / actual: Array(MutCell([1, 2])) / expected: Array(MutCell([1, 2]))


.ts output:
__assert_eq / actual:  / expected: 
__assert_eq / actual: 1 / expected: 1
__assert_eq / actual:  / expected: 
__assert_eq / actual: 2 / expected: 2
__assert_eq / actual:  / expected: 
__assert_eq / actual: 1,2 / expected: 1,2


.py output:
__assert_eq / actual: [] / expected: array('l')
__assert_eq / actual: [1] / expected: array('l', [1])
__assert_eq / actual: [] / expected: array('l')
__assert_eq / actual: [2] / expected: array('l', [2])
__assert_eq / actual: [] / expected: array('l')
__assert_eq / actual: [1, 2] / expected: array('l', [1, 2])



.fsx output:
__assert_eq / actual: [||] / expected: [||]
__assert_eq / actual: [|1|] / expected: [|1|]
__assert_eq / actual: [||] / expected: [||]
__assert_eq / actual: [|2|] / expected: [|2|]
__assert_eq / actual: [||] / expected: [||]
__assert_eq / actual: [|1; 2|] / expected: [|1; 2|]

range¶

In [ ]:
union range dim =
    | Start : dim
    | End : dim -> dim

inl range start end s =
    inl start, end =
        match start, end with
        | Start start, End fn =>
            start, s |> length |> conv |> fn
        | End start_fn, End end_fn =>
            inl len = s |> length |> conv
            start_fn len, end_fn len
    s |> slice (start |> unbox) (end |> unbox)

rust¶

vec¶

In [ ]:
nominal vec t =
    `(
        backend_switch `(()) `({}) {
            Fsharp =
                (fun () =>
                    global "#if FABLE_COMPILER\n[<Fable.Core.Erase; Fable.Core.Emit(\"Vec<$0>\")>]\n#endif\ntype Vec<'T> = class end"
                ) : () -> ()
        }
        $'' : $'Vec<`t>'
    )

from_vec¶

In [ ]:
inl from_vec forall dim el. (vec : vec el) : a dim el =
    !\\(vec, $'"fable_library_rust::NativeArray_::array_from($0)"')

from_vec_base¶

In [ ]:
inl from_vec_base forall el. (vec : vec el) : array_base el =
    !\\(vec, $'"fable_library_rust::NativeArray_::array_from($0)"')

to_vec¶

In [ ]:
inl to_vec forall t. (ab : array_base t) : vec t =
    !\\(ab, $'"$0.to_vec()"')

to_vec'¶

In [ ]:
inl to_vec' forall (t : * -> * -> *) u v. (x : t u v) : vec u =
    !\($'$"!x.to_vec()"')

to_vec''¶

In [ ]:
inl to_vec'' forall (t : * -> *) (u : * -> *) v. (x : t (u v)) : vec v =
    !\($'$"!x.to_vec()"')

to_vec'''¶

In [ ]:
inl to_vec''' forall t. (ab : array_base t) : vec t =
    !\\(ab, $'"to_vec($0)"')

vec_push¶

In [ ]:
inl vec_push forall el. (el : el) (vec : vec el) : vec el =
    inl el = join el
    inl vec = join vec
    (!\($'"true; let mut !vec = !vec"') : bool) |> ignore
    // inl vec = vec |> rust.to_mut
    (!\($'"true; !vec.push(!el)"') : bool) |> ignore
    !\($'"!vec"')

vec_reverse¶

In [ ]:
inl vec_reverse forall el. (vec : vec el) : vec el =
    inl vec = join vec
    (!\($'"true; let mut !vec = !vec"') : bool) |> ignore
    (!\($'"true; !vec.reverse()"') : bool) |> ignore
    !\($'"!vec"')

vec_retain¶

In [ ]:
inl vec_retain forall el. (fn : el -> bool) (vec : vec el) : vec el =
    inl vec = join vec
    inl fn = join fn
    (!\($'"true; let mut !vec = !vec"') : bool) |> ignore
    // inl vec = vec |> rust.to_mut
    (!\($'"true; !vec.retain(|x| !fn(x.clone()))"') : bool) |> ignore
    !\($'"!vec"')

vec_sort_by_key¶

In [ ]:
inl vec_sort_by_key forall el t. (fn : el -> t) (vec : vec el) : vec el =
    inl vec = join vec
    inl fn = join fn
    (!\($'"true; let mut !vec = !vec"') : bool) |> ignore
    // inl vec = vec |> rust.to_mut
    (!\($'"true; !vec.sort_by_key(|x| !fn(x.clone()))"') : bool) |> ignore
    !\($'"!vec"')

vec_extend¶

In [ ]:
inl vec_extend forall el. (el : vec el) (vec : vec el) : vec el =
    inl el = join el
    inl vec = join vec
    (!\($'"true; let mut !vec = !vec"') : bool) |> ignore
    // inl vec = vec |> rust.to_mut
    (!\($'"true; !vec.extend(!el)"') : bool) |> ignore
    !\($'"!vec"')

vec_collect¶

In [ ]:
inl vec_collect fn vec =
    ((;[] |> to_vec), (vec |> from_vec : _ i32 _))
    ||> am.fold fun acc x =>
        acc |> vec_extend (fn x)

vec_collect_option¶

In [ ]:
inl vec_collect_option vec =
    ((;[] |> to_vec |> Ok), (vec |> from_vec : _ i32 _))
    ||> am.fold fun acc x =>
        x
        |> resultm.unbox
        |> fun x =>
            match acc, x |> resultm.map optionm'.unbox with
            | Ok acc, Ok (Some x) => acc |> vec_extend x |> Ok
            | _, Error error => error |> Error
            | _ => acc

vec_collect_into¶

In [ ]:
inl vec_collect_into forall (c : * -> * -> *) t e.
    (x : vec (c t e))
    : c (vec t) e
    =
    !\($'"!x.into_iter().collect()"')

vec_mapi¶

In [ ]:
inl vec_mapi forall dim t u. (fn : dim -> t -> u) (ar : vec t) : vec u =
    inl fn = join fn
    inl ar = join ar
    !\($'"!ar.iter().enumerate().map(|(i, x)| !fn(i.try_into().unwrap())(x.clone())).collect::<Vec<_>>()"')

vec_map¶

In [ ]:
inl vec_map forall t u. (fn : t -> u) (ar : vec t) : vec u =
    (!\\(ar, $'"true; let _vec_map : Vec<_> = $0.into_iter().map(|x| { //"') : bool) |> ignore
    inl result = fn !\($'"x"')
    inl is_unit =
        real
            typecase u with
            | () => true
            | _ => false
    if is_unit
    then (!\($'"true; }}).collect::<Vec<_>>()"') : bool) |> ignore
    else (!\\(result, $'"true; $0 }).collect::<Vec<_>>()"') : bool) |> ignore
    !\($'"_vec_map"')

vec_map'¶

In [ ]:
inl vec_map' forall t u. (fn : t -> u) (ar : vec t) : vec u =
    !\\((ar, fn), $'"$0.into_iter().map(|x| $1(x.clone())).collect::<Vec<_>>()"')

vec_fold'¶

In [ ]:
inl vec_fold' forall t u. (fn : u -> t -> u) (init : u) (ar : vec t) : u =
    (!\\(ar, $'"true; let _vec_fold_ = $0.into_iter().fold(!init, |acc, x| { //"') : bool) |> ignore
    (!\\(fn !\($'"acc"') !\($'"x"'), $'"true; $0 })"') : bool) |> ignore
    !\($'"_vec_fold_"')

vec_for_each¶

In [ ]:
inl vec_for_each forall t. (fn : t -> ()) (ar : vec t) : () =
    (!\\((ar, fn), $'"true; $0.iter().for_each(|x| { $1(x.clone()); }); //"') : bool) |> ignore

vec_for_each'¶

In [ ]:
inl vec_for_each' forall t. (fn : t -> ()) (ar : vec t) : () =
    (!\\(ar, $'"true; $0.into_iter().for_each(|x| { //"') : bool) |> ignore
    (!\\(fn !\($'"x"'), $'$"true"') : bool) |> ignore
    (!\($'"true; }}); { //"') : bool) |> ignore

vec_for_each''¶

In [ ]:
inl vec_for_each'' forall t. (fn : t -> ()) (ar : vec t) : () =
    (!\\(ar, $'"true; $0.into_iter().for_each(|x| { //"') : bool) |> ignore
    (!\\(fn !\($'"x"'), $'$"true"') : bool) |> ignore
    (!\($'"true; }}); //"') : bool) |> ignore

vec_for_each'''¶

In [ ]:
inl vec_for_each''' forall t. (fn : t -> ()) (ar : vec t) : () =
    (!\\(ar, $'"true; $0.into_iter().for_each(|x| { //"') : bool) |> ignore
    (!\\(fn !\($'"x"'), $'$"true"') : bool) |> ignore
    (!\($'"true; }); //"') : bool) |> ignore

vec_filter¶

In [ ]:
inl vec_filter forall t. (fn : t -> bool) (ar : vec t) : vec t =
    inl fn = join fn
    inl ar = join ar
    !\($'"!ar.into_iter().filter(|x| !fn(x.clone().clone())).collect::<Vec<_>>()"')

vec_len¶

In [ ]:
inl vec_len forall t. (vec : vec t) : unativeint =
    !\\(vec, $'"$0.len()"')

vec_chunks¶

In [ ]:
inl vec_chunks forall t. (n : i32) (vec : vec t) : vec (vec t) =
    !\\(vec, $'"$0.chunks(!n).map(|x| x.into_iter().map(|x| x.clone()).collect::<Vec<_>>()).collect::<Vec<_>>()"')

slice¶

In [ ]:
nominal slice t =
    `(
        global "#if FABLE_COMPILER\n[<Fable.Core.Erase; Fable.Core.Emit(\"[$0]\")>]\n#endif\ntype Slice<'T> = class end"
        $'' : $'Slice<`t>'
    )

slice'¶

In [ ]:
nominal slice' el dim =
    `(
        backend_switch `(()) `({}) {
            Fsharp =
                (fun () =>
                    global "#if FABLE_COMPILER\n[<Fable.Core.Erase; Fable.Core.Emit(\"_\")>]\n#endif\ntype Slice'<'T> = class end"
                ) : () -> ()
        }
        $'' : $'Slice\'<`el>'
    )

slice''¶

In [ ]:
nominal slice'' el dim =
    `(
        backend_switch `(()) `({}) {
            Fsharp =
                (fun () =>
                    global "#if FABLE_COMPILER\n[<Fable.Core.Erase; Fable.Core.Emit(\"[$0; 10]\")>]\n#endif\ntype Slice'_<'T> = class end"
                ) : () -> ()
        }
        $'' : $'Slice\'_<`el>'
    )

slice_singleton¶

In [ ]:
inl slice_singleton forall dim el. (x : option el) : slice' el dim =
    match x with
    | Some x => !\($'"[!x]"')
    | None =>
        !\($'"[\\\"\\\".to_string()]"') : slice' el dim
            // emit_expr `(()) `(slice' el dim) () ($'"[@dim]"' : string) : slice' el 10
            // !\( : string) : slice' el i32 // !\($'"[]"')

slice_length¶

In [ ]:
inl slice_length forall t dim. (x : slice' t dim) : unativeint =
    !\($'"!x.len()"')

slice_range¶

In [ ]:
inl slice_range forall t dim. (start : range t) (end : range t) (s : slice' t dim) : rust.ref (slice' t dim) =
    inl len = s |> slice_length
    inl start, (end : unativeint) =
        match start, end with
        | Start start, End fn => start, len |> convert |> fn |> convert
        | End start_fn, End end_fn => len |> convert |> start_fn, len |> convert |> end_fn |> convert
    match start, end with
    | start, end when unbox end =. len => !\($'"&!s[!start..]"')
    | start, end => !\\((start, end), $'"&!s[$0..$1]"')

new_slice¶

In [ ]:
inl new_slice forall el dim. (el : el) : slice' el dim =
    !\\(el, $'"[$0; @dim]"')

as_slice¶

In [ ]:
inl as_slice forall t. (x : array_base t) : rust.ref (slice t) =
    inl x = x |> to_vec
    !\($'"!x.as_slice()"')

slice_to_vec¶

In [ ]:
inl slice_to_vec forall t. (slice : rust.ref (slice t)) : vec t =
    !\\(slice, $'"$0.iter().map(|x| *x).collect::<Vec<_>>()"')

to_le_bytes¶

In [ ]:
inl to_le_bytes forall t. (x : t) : slice' u8 8 =
    !\($'$"!x.to_le_bytes()"')

as_bytes¶

In [ ]:
inl as_bytes forall t. (x : t) : rust.ref (slice u8) =
    !\($'$"!x.as_bytes()"')

any¶

In [ ]:
inl any forall t. (fn : t -> bool) (source : array_base t) : bool =
    !\($'"!source.any(|x| !fn(x))"')

iter_collect vec¶

In [ ]:
instance iter_collect vec = fun (iter : into_iterator u) =>
    !\\(iter, $'"$0.collect::<Vec<_>>()"')

instance iter_collect'' vec = fun (iter : into_iterator (t (u v))) =>
    !\\(iter, $'"$0.collect::<Vec<_>>()"')

new_vec¶

In [ ]:
inl new_vec forall t. (items : list t) : vec t =
    inl items =
        (items, ("", 0i32))
        ||> listm.foldBack fun (x : t) (acc, i) =>
            inl x = join x
            $'"!x"' +. (if i = 0 then "" else ", ") +. acc, i + 1
        |> fst
    !\($'"vec\![" + !items + "]"')
In [ ]:
//// test
///! rust

[ 0i32; 1 ]
|> new_vec
|> sm'.format_debug'
|> sm'.from_std_string
|> _assert_eq "[0, 1]"

__assert_eq / actual: "[0, 1]" / expected: "[0, 1]"

fsharp¶

average¶

In [ ]:
inl average forall el {number}. (a : a _ el) : el =
    $'!a |> Array.average'

distinct¶

In [ ]:
inl distinct forall dim el. (a : a dim el) : a dim el =
    $'!a |> Array.distinct'

skip¶

In [ ]:
inl skip forall dim el. (n : dim) (a : a dim el) : a dim el =
    $'!a |> Array.skip !n '

skip_while¶

In [ ]:
inl skip_while forall dim el. (fn : el -> bool) (a : a dim el) : a dim el =
    $'!a |> Array.skipWhile !fn '

to_list'¶

In [ ]:
inl to_list' forall dim t. (items : a dim t) : listm'.list' t =
    $'!items |> Array.toList'

to_list_base'¶

In [ ]:
inl to_list_base' forall t. (items : array_base t) : listm'.list' t =
    $'!items |> Array.toList'
In [ ]:
//// test
///! fsharp
////! cuda //
///! rust
///! typescript
///! python

a' ;[ -3i32; 6 ]
|> to_list'
|> listm'.unbox
|> _assert_eq [ -3; 6 ]

.rs output:
__assert_eq / actual: UH0_1(-3, UH0_1(6, UH0_0)) / expected: UH0_1(-3, UH0_1(6, UH0_0))

.ts output:
__assert_eq / actual: UH0_1 (-3, UH0_1 (6, UH0_0)) / expected: UH0_1 (-3, UH0_1 (6, UH0_0))

.py output:
__assert_eq / actual: UH0_1 (-3, UH0_1 (6, UH0_0)) / expected: UH0_1 (-3, UH0_1 (6, UH0_0))


.fsx output:
__assert_eq / actual: UH0_1 (-3, UH0_1 (6, UH0_0)) / expected: UH0_1 (-3, UH0_1 (6, UH0_0))

parallel_map¶

In [ ]:
inl parallel_map forall dim el el'. (fn : el -> el') (a : a dim el) : a dim el' =
    $'!a |> Array.Parallel.map !fn '

map'¶

In [ ]:
inl map' forall dim el el'. (fn : el -> el') (a : a dim el) : a dim el' =
    $'!a |> Array.map !fn '

sort_by¶

In [ ]:
inl sort_by forall dim el. (fn : el -> _) (a : a dim el) : a dim el =
    $'!a |> Array.sortBy !fn '

sort¶

In [ ]:
inl sort forall dim el. (a : a dim el) : a dim el =
    $'!a |> Array.sort'

sort_descending¶

In [ ]:
inl sort_descending forall dim el. (a : a dim el) : a dim el =
    $'!a |> Array.sortDescending'

transpose¶

In [ ]:
inl transpose forall el. (a : array_base (array_base el)) : array_base (array_base el) =
    $'!a |> Array.transpose'

try_item¶

In [ ]:
inl try_item forall dim el. (i : i32) (a : a dim el) : option el =
    $'!a |> Array.tryItem !i ' |> optionm'.unbox