PR #28552 Fix SDPA tests

Fix SDPA tests #28552

fxmarty merged 8 commits into main from fix-sdpa-tests

skip bf16 test if not supported by device

2dafb1e8

fix

d5ca6f97

fix bis

fc72b9c3

fxmarty requested a review from

ydshieh 2 years ago

fxmarty requested a review from

ArthurZucker 2 years ago

amyeroberts commented on 2024-01-17

use is_torch_bf16_available_on_device

381ceb9c

fxmarty requested a review from

amyeroberts 2 years ago

amyeroberts approved these changes on 2024-01-17

use is_torch_fp16_available_on_device

c033e882

ydshieh commented on 2024-01-17

ydshieh approved these changes on 2024-01-17

fix & use public llama

90b794db

use 1b model

39f9d24d

fix flacky test

9155e8d3

fxmarty merged 2c1eebc1 into main 2 years ago

fxmarty deleted the fix-sdpa-tests branch 2 years ago

Reviewers

ydshieh

amyeroberts

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone