Commit Graph
7195 Commits
Author SHA1 Message Date
qazal dccfcbe068 this fixes test_train_mnist memory, breaks everything else 2024-12-07 17:38:20 +02:00
qazal 4f98464bfc iterate on test_real_world 2024-12-07 15:29:51 +02:00
qazal 1722b2dd6b fix test_prefer_half_buffer 2024-12-07 15:20:20 +02:00
qazalandGitHub 644176bff4 Merge branch 'master' into delete_lazy 2024-12-07 15:16:41 +02:00
qazal cb6c72f577 storing const is fine 2024-12-07 14:55:18 +02:00
qazalandGitHub 6be388be86 failing test for const folding breaking indexing [pr] (#8103) 2024-12-07 19:55:02 +08:00
qazal b7b73316df Revert VALID early folding
This reverts commit 4074f52317.
2024-12-07 13:24:35 +02:00
qazal b0fb616d4a Revert "more const folding"
This reverts commit 2a7d258a2b.
2024-12-07 13:24:00 +02:00
qazal cfda7b6694 Revert "experiment with symbolic here"
This reverts commit cb87d61f7a.
2024-12-07 13:23:59 +02:00
qazal cb87d61f7a experiment with symbolic here 2024-12-07 13:21:49 +02:00
nimlgenandGitHub 8b1fa9cb7d nv hcq queue touchups (#8102) 2024-12-07 14:09:38 +03:00
qazal 2a7d258a2b more const folding 2024-12-07 13:04:28 +02:00
qazal 25f5b336ef Merge remote-tracking branch 'origin/master' into delete_lazy 2024-12-07 12:42:47 +02:00
qazalandGitHub 4074f52317 VALID early folding (#8100)
* fold valid

* :)

* fix test_verify_ast

* keep symbolic working
2024-12-07 18:37:47 +08:00
qazal 39f2fd8787 more diff pruning 2024-12-07 11:59:06 +02:00
qazal d5eb68482b Merge remote-tracking branch 'origin/master' into delete_lazy 2024-12-07 11:56:37 +02:00
qazalandGitHub 07b6d5cf63 assign early folding (#8093)
* assign early folding [pr]

* move to to_si

* -

* fix generate_dataset

* diff too big

* no recreation, no diff

* gzip

* new sops from tiny10

* final try
2024-12-07 17:02:55 +08:00
George HotzandGitHub 00ac0db9d4 np tensors have the memory from numpy in compile3 [pr] (#8098) 2024-12-07 14:01:51 +08:00
George HotzandGitHub 22feb3a2f1 move copy into the JIT for openpilot compile3 (#7937)
* move copy into the JIT, test fails

* ahh, prune was the issue
2024-12-07 13:26:26 +08:00
leopfandGitHub 0ed731b5ea torch_load with Tensors (#8037)
* torch_load with Tensors

* remove passthrough_reset + use accept_filename

* Revert "remove passthrough_reset"

* version note

* cleanup
2024-12-07 09:55:41 +08:00
chenyuandGitHub 2d321646b8 default tensors to int32 in test_ops (#8097)
torch defaults to int64 but we care more about int32 anyway. remove skipped tests due to int64 not supported
2024-12-06 20:33:36 -05:00
chenyuandGitHub e9692de42b don't FUZZ_ALL_ACTIONS in fuzz_linearizer.py (#8096)
mostly for speed, this is just making sure the script runs
2024-12-06 17:22:17 -05:00
chenyuandGitHub 564b3a3e1b onnx Bitwise ops (#8095)
free stuff!
2024-12-06 16:58:09 -05:00
qazalandGitHub a97b8fa3c5 maskless const can lower without valid, p1 [pr] (#8094) 2024-12-06 23:21:19 +02:00
mesozoic-eggandGitHub aaf2379f97 remove ordered parents, seems like dead code [pr] (#8092)
* remove ordered parents, seems like dead code

* no need to dedup
2024-12-06 16:19:37 -05:00
qazal 4fd1fdaf43 extra 2024-12-06 22:07:48 +02:00
qazal c6b47b4c63 oh bind unbinds way too early 2024-12-06 22:03:20 +02:00
qazal 6507e7f402 Merge remote-tracking branch 'origin/master' into delete_lazy 2024-12-06 20:58:56 +02:00
nimlgenandGitHub e180a31c5e tiny metal cleanup (#8089)
* tiny metal cleanup

* cast

* sry
2024-12-06 21:44:32 +03:00
chenyuandGitHub d000c08f04 fix return type of Tensor.pow (#8091)
int to power of int should return int etc, it hints that we would like to have Ops.POW
2024-12-06 13:38:29 -05:00
qazalandGitHub 1ea4dc9565 big graph init conceptual cleanup [pr] (#8090)
* keep Ops.BUFFER naming consistent [pr]

* big graph init conceptual cleanup [pr]

* make everything pass through

* pylint doesn't complain now
2024-12-06 20:07:00 +02:00
5184410fc3 combine get inputs and type_parse function in onnx [fixed] (#8081)
* 1 is simpler than 2

* variable name

* change error wording

* shapes for sequence type must be homogeneous

* bug fix for model benchmark

* fix comments too

---------

Co-authored-by: chenyu <[email protected]>
2024-12-06 12:34:47 -05:00
qazal 97e188a2a7 diff pruning 2 2024-12-06 18:49:33 +02:00
qazal 07b22370e2 diff pruning 2024-12-06 18:44:30 +02:00
qazal 38d9efc219 copy is one thing 2024-12-06 18:37:44 +02:00
nimlgenandGitHub d1282da7e8 hcq bump alloc (#8078)
* hcq bump alloc

* hm

* nv

* typo
2024-12-06 19:19:04 +03:00
qazal 3f8193d747 Merge remote-tracking branch 'origin/master' into delete_lazy 2024-12-06 17:40:52 +02:00
qazalandGitHub df84dc6444 unrelated test fixups from delete_lazy [pr] (#8088)
* unrelated test fixups from delete_lazy [pr]

* fine if it's scheduled later
2024-12-06 17:31:02 +02:00
0b7c44677d Fix uint8 cast underflow (#6305)
* hacky fix for cast

* only float to uint8

* limit to float -> uint8

* touchup alu cast test

* improve tests and support more float to unsigned casts

* del one repeated test

* del 1 more repeated test

* try removing expected failure test

* hmmm try 1 more

* skip tests for flakiness

* uint64 super flaky

* clean up

* grammar

* just match numpy

* why is CI numpy different from local numpy

* increase verbosity

* try

* try2

* try3

* try4

* yeah idk

* new direction

* try again

* just don't support uint32 and uint64

* done?

* oops

* comment

* documentation

* it is what it is

---------

Co-authored-by: chenyu <[email protected]>
2024-12-06 10:25:03 -05:00
qazal 57f2f04544 Merge remote-tracking branch 'origin/master' into delete_lazy 2024-12-06 16:58:05 +02:00
Ahmed HarmoucheandGitHub f3983f6743 Move efficientnet example (#8087) 2024-12-06 15:48:16 +01:00
qazalandGitHub 7dbd166227 skip test_schedule_mem_used_with_inputs [pr] (#8086) 2024-12-06 16:44:34 +02:00
qazalandGitHub 0356657ced move view_supported_devices to device [pr] (#8085) 2024-12-06 16:44:15 +02:00
Ahmed HarmoucheandGitHub fad3eaa35e Use atomicLoad builtin when loading atomic type (#8084) 2024-12-06 15:33:11 +01:00
qazalandGitHub 79966fade0 free up lines for const_arg [pr] (#8083) 2024-12-06 16:28:51 +02:00
qazal 79feb959e6 cleanup the diff in test/ 1 2024-12-06 16:06:38 +02:00
Ahmed HarmoucheandGitHub ba35c4138b Use matching JS TypedArray for buffer dtype (#8080) 2024-12-06 14:52:23 +01:00
a684d72e55 add ceil_mode for avg_pool and max_pool (#7579)
* wip pool

* check CI for remove alternative implementation

* Revert "check CI for remove alternative implementation"

This reverts commit 7b1bb900e5.

* fix test

* tests tests tests

* slap a resolve on it

* fix comment

* a little simpler pool

* check CI for removal again

* Revert "check CI for removal again"

This reverts commit be798b7857.

* small

* update

* some ez tests

* english

* clean up code

* fix ruff

* how did I +25 lines?

* small clean ups

* moar clean ups

* try test_avgpool2d_failure2 in CI

* final clean up

* exclude bug fix

* avg underscore pool

* no more edge case stuff

* add better comments for explanation

* add test cases for decreasing end padding

* address feedback

* improve test coverage

* tiny more polish as we wait for lines :D

* more readable code ordering

* add to documentation

* oops

* set to False instead

---------

Co-authored-by: chenyu <[email protected]>
2024-12-06 08:34:14 -05:00
qazal 52f6bd70f1 Merge remote-tracking branch 'origin/master' into delete_lazy 2024-12-06 15:06:23 +02:00
qazal 4ba77164bd this is fine with buffer view 2024-12-06 15:06:21 +02:00