Skip to content

Commit 1acec6f

Browse files
authored
Fix compilation errors. (#724)
1 parent 1ac9959 commit 1acec6f

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

k2/csrc/array_test.cu

+6-4
Original file line numberDiff line numberDiff line change
@@ -264,16 +264,18 @@ void TestArray2() {
264264
Array1<T> arr1(context, data);
265265
Array2<T> array(arr1, kDim0, kDim1);
266266
Array2<T> array_copy = array.Generic().template Specialize<T>();
267-
K2_CHECK_EQ(true, Equal(array, array_copy));
268-
K2_CHECK_EQ(array.Data(), array_copy.Data());
267+
EXPECT_TRUE(Equal(array, array_copy));
268+
EXPECT_EQ(array.Data(), array_copy.Data());
269269

270270
const Array2<T> &array_copy_const = array;
271271
const Array2<T> &array_copy_const2 = array_copy_const.Generic().template Specialize<T>();
272-
K2_CHECK_EQ(array_copy_const.Data(), array_copy.Data());
272+
EXPECT_EQ(array_copy_const.Data(), array_copy.Data());
273+
EXPECT_EQ(array_copy_const2.Data(), array_copy.Data());
273274

274275
const Array1<T> &array1_copy_const = arr1;
275276
const Array1<T> &array1_copy_const2 = array1_copy_const.Generic().template Specialize<T>();
276-
K2_CHECK_EQ(array1_copy_const.Data(), array1_copy.Data());
277+
EXPECT_EQ(array1_copy_const.Data(), arr1.Data());
278+
EXPECT_EQ(array1_copy_const2.Data(), arr1.Data());
277279

278280

279281
auto cpu_array = array.To(cpu);

k2/csrc/ragged.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -398,7 +398,7 @@ struct Ragged {
398398
Ragged<Any> &Generic() {
399399
return *reinterpret_cast<Ragged<Any>*>(this);
400400
}
401-
Ragged<Any> &Generic() const {
401+
const Ragged<Any> &Generic() const {
402402
return *reinterpret_cast<const Ragged<Any>*>(this);
403403
}
404404

0 commit comments

Comments
 (0)