accelerate
Prefer `is_torch_tensor` over `hasattr` for torch.compile.
#2387

Merged

Prefer `is_torch_tensor` over `hasattr` for torch.compile. #2387

muellerzr merged 8 commits into huggingface:main from PhilJd:main

muellerzr commented on 2024-01-30

PhilJd force pushed from 443963e7 to 1f0f535a 2 years ago

muellerzr approved these changes on 2024-02-08

muellerzr commented on 2024-02-12

muellerzr approved these changes on 2024-02-12

Prefer `is_torch_tensor` over `hasattr` for `torch.compile`.

ef6d8d69

Fix torch.device("npu").

816f9508

Remove unittest.main execution.

0392a724

Fix style broken by merge save.

ff03d608

Import operations functions directly.

947792f0

fix style

d6a5a15c

Fix imports attempt 2.

3aa55ba3

PhilJd force pushed from 18825c6c to 3aa55ba3 2 years ago

BenjaminBossan commented on 2024-02-13

Re-raise error if no NPU available.

f9dd02ec

PhilJd commented on 2024-02-13

BenjaminBossan approved these changes on 2024-02-14

muellerzr merged aba3b8c7 into main 2 years ago

Reviewers

BenjaminBossan

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone