This essentially just made DeviceArray and DeviceScalar into distinct Protocol classes (not just DeviceScalar = Any) and added a little check in dataclass_array_container.
The motivation is to maybe eventually replace boxtree.DeviceDataRecord with some standard dataclasses that support arraycontext.
This essentially just made
DeviceArray
andDeviceScalar
into distinctProtocol
classes (not justDeviceScalar = Any
) and added a little check indataclass_array_container
.The motivation is to maybe eventually replace
boxtree.DeviceDataRecord
with some standard dataclasses that supportarraycontext
.