[dipu] Fix copy_ fallback of topsrider. (DeepLink-org#477)

DeepLink-org · Dec 8, 2023 · ea0d7a5 · ea0d7a5
1 parent 6b29468
commit ea0d7a5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml b/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml
@@ -2175,6 +2175,7 @@
 
 # vendor who has no fully implemented diopi and proper fallback DIPUCopy sub-class
 - schema: copy_(Tensor(a!) self, Tensor src, bool non_blocking=False) -> Tensor(a!)
+  custom_fallback: True
   dummy_call_diopi: True
   custom_code_at_the_beginning: |
     return custom_fallback_dipu_copy_(self, src, non_blocking);