Skip to content

Commit

Permalink
feat(dipu): support ForceFallbackToCPU (#929)
Browse files Browse the repository at this point in the history
* feat: support ForceFallbackToCPU

* fix: use bool instead of string

* feat: init out as empty if sizes not equal

* fix: throw error when out is cpu tensor

* fix: replace list and tuple for Python3.8

* doc: add comments

* refactor: rename enable_fallback_to_cpu
  • Loading branch information
wiryls authored Oct 10, 2024
1 parent aec0e71 commit 0caf4c8
Show file tree
Hide file tree
Showing 7 changed files with 549 additions and 331 deletions.
1 change: 0 additions & 1 deletion dipu/SupportedDiopiFunctions.txt
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,6 @@ diopiBitwiseOrInp
diopiBitwiseOrInpScalar
diopiBitwiseOrScalar
diopiBmm
diopiCastDtype
diopiCat
diopiCdist
diopiCdistBackward
Expand Down
Loading

0 comments on commit 0caf4c8

Please sign in to comment.