提交 6e04e580 编写于 作者: W wanghaoshuang

Change input data type to int64_t in unitest and GPU kernel

上级 77cf21e5
...@@ -39,8 +39,8 @@ __global__ void FillFirstColumn(T* dist, const int M, const int N) { ...@@ -39,8 +39,8 @@ __global__ void FillFirstColumn(T* dist, const int M, const int N) {
} }
template <typename T> template <typename T>
__global__ void Levenshtein(T* dist, const int* x1, const int* x2, const int M, __global__ void Levenshtein(T* dist, const int64_t* x1, const int64_t* x2,
const int N, const int start) { const int M, const int N, const int start) {
int idx = blockDim.x * blockIdx.x + threadIdx.x; int idx = blockDim.x * blockIdx.x + threadIdx.x;
int offset = N; int offset = N;
int index = start + idx * offset; int index = start + idx * offset;
......
...@@ -51,8 +51,8 @@ class TestEditDistanceOp(OpTest): ...@@ -51,8 +51,8 @@ class TestEditDistanceOp(OpTest):
def setUp(self): def setUp(self):
self.op_type = "edit_distance" self.op_type = "edit_distance"
normalized = False normalized = False
x1 = np.array([[0, 12, 3, 5, 8, 2]]).astype("int32") x1 = np.array([[0, 12, 3, 5, 8, 2]]).astype("int64")
x2 = np.array([[0, 12, 4, 7, 8]]).astype("int32") x2 = np.array([[0, 12, 4, 7, 8]]).astype("int64")
x1 = np.transpose(x1) x1 = np.transpose(x1)
x2 = np.transpose(x2) x2 = np.transpose(x2)
x1_lod = [0, 1, 5] x1_lod = [0, 1, 5]
...@@ -79,8 +79,8 @@ class TestEditDistanceOpNormalized(OpTest): ...@@ -79,8 +79,8 @@ class TestEditDistanceOpNormalized(OpTest):
def setUp(self): def setUp(self):
self.op_type = "edit_distance" self.op_type = "edit_distance"
normalized = True normalized = True
x1 = np.array([[0, 10, 3, 6, 5, 8, 2]]).astype("int32") x1 = np.array([[0, 10, 3, 6, 5, 8, 2]]).astype("int64")
x2 = np.array([[0, 10, 4, 6, 7, 8]]).astype("int32") x2 = np.array([[0, 10, 4, 6, 7, 8]]).astype("int64")
x1 = np.transpose(x1) x1 = np.transpose(x1)
x2 = np.transpose(x2) x2 = np.transpose(x2)
x1_lod = [0, 1, 3, 6] x1_lod = [0, 1, 3, 6]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册