Device agnostic testing #25870
ydshieh
merged
19 commits into
huggingface:main
from
graphcore:device-agnostic-testing
adds agnostic decorators and availability fns
1327ab78
renaming decorators and fixing imports
98bb8fc9
updating some representative example tests
9f4c9efa
wip device agnostic functions
d09ea4b0
lru cache to device checking functions
2c9afb4a
adds `TRANSFORMERS_TEST_DEVICE_SPEC`
4d12550f
comments `TRANSFORMERS_TEST_DEVICE_SPEC` code
31c61eda
extra checks on device name
686d8f21
`make style; make quality`
7c9e892a
updates default functions for agnostic calls
c961bd63
ydshieh
commented
on 2023-09-08
applies suggestions from review
a582f933
adds `is_torch_available` guard
bc65a0e9
Add spec file to docs, rename function dispatch names to backend_*
d0e5443d
ydshieh
commented
on 2023-10-04
add backend import to docs example for spec file
fe2b4981
change instances of to
8492352b
Merge branch 'main' of https://github.com/graphcore/transformers-fork…
dd5ddcbe
Move register backend to before device check as per @statelesshz changes
7c302ba5
make style
a1230d45
ydshieh
approved these changes
on 2023-10-24
LysandreJik
commented
on 2023-10-24
fxmarty
approved these changes
on 2023-10-24
LysandreJik
approved these changes
on 2023-10-24
make opt test require fp16 to run
d696b55c
Reviewers
Assignees
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub