chenyuandGitHub b282a45e39 fix direct store float4 with same vin (#3652) ...

In a kernel that stores expanded value, the vin of float4 can come from same source, and we only remove once in that case.

2024-03-07 18:11:50 -05:00

..

codegen

fix direct store float4 with same vin (#3652)

2024-03-07 18:11:50 -05:00

features

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

nn

Improve type hints for optimizer (#3583)

2024-03-02 07:35:44 -08:00

renderer

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

runtime

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

shape

Fix symbolic negative floordiv (#3594)

2024-03-03 11:40:52 -05:00

__init__.py

move graph.py and jit.py into features (#3376)

2024-02-12 17:34:34 +01:00

device.py

bring ptx back (#3623)

2024-03-06 13:34:21 -08:00

dtype.py

Rename .sz to .count on DType (#3413)

2024-02-15 15:03:49 +01:00

helpers.py

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

lazy.py

lazy.py: cache consts (#3577)

2024-03-02 03:50:05 -08:00

mlops.py

change frontend reduce API to use axis (#3460)

2024-02-21 12:26:17 +01:00

ops.py

fix typing FlopCounter.flops can be sint (#3646)

2024-03-07 12:49:17 -05:00

realize.py

update kernel dataset to exclude the disktensor ones (#3651)

2024-03-07 17:35:19 -05:00

tensor.py

fix Tensor.to preserves grad.data (#3636)

2024-03-06 21:44:49 -05:00