From a7f6701b37af5e7f486d480d050052ac975f47a6 Mon Sep 17 00:00:00 2001
From: jayshu <shuhaojie@huawei.com>
Date: Mon, 5 Feb 2024 16:08:23 +0800
Subject: [PATCH] add the input contiguous

---
 ads/common/ops/npu_multi_scale_deformable_attn_function.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ads/common/ops/npu_multi_scale_deformable_attn_function.py b/ads/common/ops/npu_multi_scale_deformable_attn_function.py
index 206ee4b6..4a8ed145 100644
--- a/ads/common/ops/npu_multi_scale_deformable_attn_function.py
+++ b/ads/common/ops/npu_multi_scale_deformable_attn_function.py
@@ -19,7 +19,7 @@ class MultiScaleDeformableAttnFunction(Function):
         value, shape, offset, locations, weight = ctx.saved_tensors
         grad_value, grad_sample_loc, grad_atten_weight = ads_c.multi_scale_deformable_attn_grad(value, shape, offset,
                                                                                                 locations, weight,
-                                                                                                grad_output)
+                                                                                                grad_output.contiguous())
         return grad_value, None, None, grad_sample_loc, grad_atten_weight
 
 
-- 
Gitee