This repository was archived by the owner on Feb 7, 2024. It is now read-only.
This repository was archived by the owner on Feb 7, 2024. It is now read-only.
Trace cache: add shape, dtype, etc for key equality (but not the hash) #9
Description
The TorchScript backend uses: t.scalar_type(), t.device(), t.sizes(), t.strides(), t.requires_grad(), t.is_contiguous()
In theory it can specialize traces for this data, so we need to take them into account when looking up traces in the cache.