tinygrad/test
chenyu 8a8aed23d2
test dtypes of return values of cumsum, argmax/min, multinomial (#2933)
* test dtypes of return values of cumsum, argmax/min, multinomial

cumsum behaves like sum, and functions that return an index return in dtypes.default_int

* because webgpu is different
2023-12-25 11:33:17 -05:00
..
external hotfix fix coder. RMSNorm cannot have float16 input (#2932) 2023-12-25 02:28:11 -05:00
extra Remove pytest markers (#2831) 2023-12-18 18:53:28 -05:00
imported new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
models hotfix fix coder. RMSNorm cannot have float16 input (#2932) 2023-12-25 02:28:11 -05:00
unit apply the multitensor optimizations in lazy.py (#2901) 2023-12-21 13:55:49 -08:00
web fast path for copy (#2548) 2023-12-01 11:34:47 -08:00
Dockerfile Docker fix (#1039) 2023-06-25 10:38:58 -07:00
__init__.py All devices are equal! (#196) 2020-12-15 23:44:08 -08:00
helpers.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_assign.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_conv.py update pytest marks and CI test filters (#2587) 2023-12-03 15:20:44 -05:00
test_conv_shapetracker.py fix some tests related to JitItem (#2279) 2023-11-11 23:00:35 -05:00
test_copy_speed.py Disk shm refactor (#2912) 2023-12-22 09:23:37 -08:00
test_custom_function.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_dtype.py test dtypes of return values of cumsum, argmax/min, multinomial (#2933) 2023-12-25 11:33:17 -05:00
test_dtype_alu.py cleanup test_dtype_alu (#2919) 2023-12-22 17:29:31 -05:00
test_fusion_op.py hotfix: compare on ids 2023-12-20 23:47:50 -08:00
test_gc.py fix up eye and fix gc test 2023-02-27 06:53:18 -08:00
test_hip_rdna3.py HIP cleanups (#2843) 2023-12-18 21:09:32 -08:00
test_image_dtype.py s/lazydata.realized/lazydata.base.realized/g (#2914) 2023-12-22 14:45:13 -05:00
test_jit.py lazy tests (#2796) 2023-12-16 08:24:21 -08:00
test_kernel_cache.py move device to device.py (#2466) 2023-11-27 11:34:37 -08:00
test_lazybuffer.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_lazyop.py Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_linearizer.py s/lazydata.realized/lazydata.base.realized/g (#2914) 2023-12-22 14:45:13 -05:00
test_linearizer_failures.py update dtypes (#2872) 2023-12-20 15:04:02 -05:00
test_masked_st.py multitensor start (#2676) 2023-12-07 17:07:05 -08:00
test_net_speed.py Remove pytest markers (#2831) 2023-12-18 18:53:28 -05:00
test_nn.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_ops.py Einsum space fix (#2927) 2023-12-24 01:23:27 -05:00
test_optim.py remove realize from optimizer (#2880) 2023-12-20 16:42:41 -08:00
test_randomness.py Fixed Tensor.randint() not accepting tuple shapes (#2923) 2023-12-24 20:32:26 -05:00
test_sample.py enable test_sample for all backend (#2593) 2023-12-03 17:20:27 -05:00
test_schedule.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_search.py new style device (#2530) 2023-11-30 17:07:16 -08:00
test_specific_conv.py Non fp32 math (#2264) 2023-12-03 13:45:49 -08:00
test_speed_v_torch.py Remove pytest markers (#2831) 2023-12-18 18:53:28 -05:00
test_symbolic_jit.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_symbolic_ops.py new lazy, benchmark (#2878) 2023-12-20 14:33:21 -08:00
test_symbolic_shapetracker.py tests from new lazy branch (#2774) 2023-12-14 23:06:39 -08:00
test_tensor.py bugfix Tensor.item when it's unbased (#2913) 2023-12-22 13:50:06 -05:00
test_tensor_data.py masked shapetrackers (#2657) 2023-12-06 11:22:26 -08:00
test_to_numpy.py Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_uops.py TernaryOps.WHERE has vin[0] as bool and BinaryOps.CMPLT always outputs bool (#2782) 2023-12-15 14:51:51 -05:00
test_winograd.py get_lazyops() -> lazyops (#2884) 2023-12-20 18:04:49 -08:00
test_zero_copy.py s/lazydata.realized/lazydata.base.realized/g (#2914) 2023-12-22 14:45:13 -05:00