mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
tinygrad/test
History
Marcello Fuschi b8d460d203
 Add Tensor.multinomial (#2295) 
* add Tensor.multinomial only with replacement

* add support for 2D input in Tensor.multinomial

* fix multinomial output shape

* allow passing replacement=False to Tensor.multinomial when num_samples=1

* improve tests for Tensor.multinomial

* fix edge case in Tensor.multinomial

* Tensor.multinomial no more staticmethod
2023-11-15 11:38:39 -08:00
..
external JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
extra Replace (getenv("CI", "") != "") with helpers.CI (#2213) 2023-11-03 15:20:44 -07:00
models JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
unit move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
__init__.py All devices are equal! (#196) 2020-12-15 23:44:08 -08:00
Dockerfile Docker fix (#1039) 2023-06-25 10:38:58 -07:00
helpers.py perf: lazyop as dataclass (#1603) 2023-10-25 17:54:30 -04:00
test_allocators.py For cuda get current free space from device, and retry alloc failures (#2197) 2023-11-09 15:53:50 -08:00
test_assign.py move optimize_local_size (#2221) 2023-11-05 21:00:52 -08:00
test_conv.py fix wino mask upcast calculation (#2057) 2023-10-18 16:54:48 -07:00
test_conv_shapetracker.py fix some tests related to JitItem (#2279) 2023-11-11 23:00:35 -05:00
test_custom_function.py JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
test_dtype.py Internal cast 2 with more tests (#2257) 2023-11-10 10:42:39 -08:00
test_gc.py fix up eye and fix gc test 2023-02-27 06:53:18 -08:00
test_jit.py JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
test_kernel_cache.py move all to compile api (#2203) 2023-11-01 23:01:32 -07:00
test_lazybuffer.py examples of new GPT2 and JIT change (#2261) 2023-11-10 15:07:02 -05:00
test_lazyop.py make self referential AST fast too (#2278) 2023-11-15 07:12:07 -08:00
test_linearizer.py examples of new GPT2 and JIT change (#2261) 2023-11-10 15:07:02 -05:00
test_linearizer_failures.py explicit opts for test_linearizer_failures (#2299) 2023-11-14 11:52:38 -05:00
test_net_speed.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_nn.py fix wino mask upcast calculation (#2057) 2023-10-18 16:54:48 -07:00
test_ops.py allow 0-dim array to broadcast into zero shape tensor (#2315) 2023-11-15 13:12:21 -05:00
test_optim.py use class Foo: instead of class Foo(): (#1797) 2023-09-06 12:20:25 -07:00
test_randomness.py Add Tensor.multinomial (#2295) 2023-11-15 11:38:39 -08:00
test_sample.py remove real_offset (#2234) 2023-11-07 17:30:53 -08:00
test_schedule.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
test_search.py simple runtime args (#2211) 2023-11-03 12:31:29 -07:00
test_specific_conv.py move device to ops (#1646) 2023-08-23 08:30:17 -07:00
test_speed_v_torch.py move optimize_local_size (#2221) 2023-11-05 21:00:52 -08:00
test_symbolic_jit.py JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
test_symbolic_ops.py JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
test_symbolic_shapetracker.py remove real_offset (#2234) 2023-11-07 17:30:53 -08:00
test_tensor.py Tensor(None) is Tensor([]) (#2316) 2023-11-15 13:49:18 -05:00
test_uops.py JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
test_webgpu.js Enable Multi-Output Export (#2179) 2023-10-30 18:42:26 -07:00
test_winograd.py print test durations and add speed (#2107) 2023-10-18 13:46:42 -07:00