Skip to content
This repository has been 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

Open
nunoplopes opened this issue Oct 6, 2021 · 0 comments

Comments

@nunoplopes
Copy link
Owner

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.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant