42 commits

Author	SHA1	Message	Date
chenyu	c33b767407	bring back test and torch backend change for unique const (#16403)	2026-05-27 15:16:08 -04:00
chenyu	945ed4f689	revert const unique changes (#16395)	2026-05-27 00:06:41 -04:00
chenyu	0b88827482	remove CONST(UNIQUE) (#16383)	2026-05-26 14:45:22 -04:00
chenyu	d861c50dce	remove unique_const (#16382)	2026-05-26 13:53:31 -04:00
chenyu	9b00defc8c	Revert "remove unique_const (#16372)" (#16380) ... This reverts commit 09019d6761.	2026-05-26 12:30:07 -04:00
chenyu	09019d6761	remove unique_const (#16372) ... * remove unique_const * fix SDWA thing * that?	2026-05-26 12:18:03 -04:00
chenyu	beea4633fc	UOp.clone [pr] (#16295) ... generates the store after structure	2026-05-20 17:47:49 -04:00
chenyu	32138c2418	svd to mixin (#16175)	2026-05-12 22:29:01 -04:00
chenyu	420a08c6d1	qr to mixin (#16173)	2026-05-12 21:23:25 -04:00
chenyu	072db9924c	div to mixin (#16078) ... also deleted idiv method	2026-05-07 12:52:37 -04:00
chenyu	516b00e286	mod and fmod to mixin (#16077)	2026-05-07 12:13:39 -04:00
chenyu	654e611a29	_bits_to_rand to mixin (#15972)	2026-04-29 13:47:25 -04:00
chenyu	fb188c3c23	UOp.bitcast noop early return (#15968) ... matches Tensor	2026-04-29 09:41:40 -04:00
chenyu	c4bea54e9c	_threefry_random_bits to mixin (#15959) ... start RandMixin	2026-04-28 19:13:57 -04:00
chenyu	77f9125c21	move Tensor.pad to OpMixin (#15946)	2026-04-27 16:56:04 -04:00
chenyu	fe38d6de94	_pad_circular and _pad_reflect_replicate to mixin (#15944)	2026-04-27 16:07:05 -04:00
chenyu	b501ba3e42	nll_loss to mixin (#15918)	2026-04-24 15:50:31 -04:00
chenyu	2f9fdb4a37	scatter to mixin (#15917)	2026-04-24 15:37:37 -04:00
chenyu	03a7604f76	sort argsort topk allclose to mixin (#15910)	2026-04-24 10:20:46 -04:00
chenyu	c24da99d56	avg_pool2d, max_pool2d to mixin (#15903) ... * avg_pool2d, max_pool2d to mixin * fix * just dtype * that	2026-04-23 23:36:17 -04:00
chenyu	08d9106c9f	scatter_reduce and sparse_categorical_crossentropy to mixin (#15902) ... also use `.ne` to fix `# type: ignore[comparison-overlap]`	2026-04-23 21:06:36 -04:00
chenyu	8cc2c69e21	fix isclose mixin (#15898) ... use `.eq` instead of `==`	2026-04-23 20:40:43 -04:00
chenyu	782bc6aece	broadcast in ElementwiseMixin.div [pr] (#15897)	2026-04-23 16:02:43 -04:00
chenyu	11c197955b	interpolate and cross_entropy to mixin (#15895)	2026-04-23 14:59:45 -04:00
chenyu	f0dbc68aa9	gather to mixin (#15891)	2026-04-23 14:00:57 -04:00
chenyu	87223f870e	logcumsumexp, argmax, argmin, sequential to mixin (#15890)	2026-04-23 12:10:42 -04:00
chenyu	1fc4b3788c	cummax/cummin to mixin (#15877)	2026-04-22 21:25:39 -04:00
chenyu	684e95e1d4	UOp binary op broadcasts dtype (#15875) ... * UOp binary op broadcasts dtype matches Tensor * fix * fix?	2026-04-22 20:37:19 -04:00
chenyu	3821e442eb	_one_hot_along_dim and one_hot to mixin (#15861)	2026-04-21 20:24:38 -04:00
chenyu	75ee51a446	triu tril _tri to mixin (#15857)	2026-04-21 17:10:55 -04:00
chenyu	1946ae8b51	linspace and eye to mixin (#15854)	2026-04-21 15:58:03 -04:00
chenyu	86ceb3bd6b	arange to mixin (#15852)	2026-04-21 13:00:19 -04:00
chenyu	420e4c4673	zeros, ones, invalids to mixin (#15850)	2026-04-21 11:53:08 -04:00
chenyu	d08b5d0a3b	full to mixin (#15840) ... with unique_const	2026-04-21 10:53:43 -04:00
chenyu	cabc347066	conv2d and conv_transpose2d to mixin (#15838) ... * conv2d and conv_transpose2d to mixin * cleanup	2026-04-20 18:10:06 -04:00
chenyu	b01704444b	einsum to ReduceMixin (#15833)	2026-04-20 11:49:24 -04:00
chenyu	2a5a6236ac	UOp.empty and UOp.empty_like (#15816) ... * UOp.empty and UOp.empty_like Tensor.empty and Tensor.empty_like use these, and removed _buffer_like * import line	2026-04-19 16:01:01 -04:00
chenyu	1fac03ce54	softmax and friends to mixin (#15778) ... with detach now	2026-04-16 23:03:37 -04:00
chenyu	70883a6950	cat the stack to mixin (#15715)	2026-04-13 18:44:39 -04:00
chenyu	ac41f15fc1	cumsum to mixin (#15712) ... built on top of getitem	2026-04-13 15:06:08 -04:00
chenyu	931d6cc62a	basic getitem to mixin (#15697) ... * basic getitem to mixin * cleanup * fix * cleanup	2026-04-13 13:04:36 -04:00
chenyu	f7ff480fa6	start mixin getitem tests (#15695) ... goal is to make Tensor[idx].uop equal to Tensor.uop[idx]	2026-04-12 18:54:33 -04:00