3030
3131namespace cinn ::frontend {
3232
33- TEST (GemmRwriter , BatchedTransLeft) {
33+ TEST (GemmRewriter , BatchedTransLeft) {
3434 if (!cinn::runtime::IsCompiledWithCUDA ()) {
3535 return ;
3636 }
@@ -54,7 +54,7 @@ TEST(GemmRwriter, BatchedTransLeft) {
5454 CompareResult (&program, target, input_ids, {out->id }, 1 , passes, 123 , true );
5555}
5656
57- TEST (GemmRwriter , BatchedTransRight) {
57+ TEST (GemmRewriter , BatchedTransRight) {
5858 if (!cinn::runtime::IsCompiledWithCUDA ()) {
5959 return ;
6060 }
@@ -78,7 +78,7 @@ TEST(GemmRwriter, BatchedTransRight) {
7878 CompareResult (&program, target, input_ids, {out->id }, 1 , passes, 123 , true );
7979}
8080
81- TEST (GemmRwriter , BatchedTransTwo) {
81+ TEST (GemmRewriter , BatchedTransTwo) {
8282 if (!cinn::runtime::IsCompiledWithCUDA ()) {
8383 return ;
8484 }
@@ -103,7 +103,7 @@ TEST(GemmRwriter, BatchedTransTwo) {
103103 CompareResult (&program, target, input_ids, {out->id }, 2 , passes, 123 , true );
104104}
105105
106- TEST (GemmRwriter , BatchedNoTrans) {
106+ TEST (GemmRewriter , BatchedNoTrans) {
107107 if (!cinn::runtime::IsCompiledWithCUDA ()) {
108108 return ;
109109 }
@@ -126,7 +126,7 @@ TEST(GemmRwriter, BatchedNoTrans) {
126126 CompareResult (&program, target, input_ids, {out->id }, 0 , passes, 123 , true );
127127}
128128
129- TEST (GemmRwriter , TransLeft) {
129+ TEST (GemmRewriter , TransLeft) {
130130 if (!cinn::runtime::IsCompiledWithCUDA ()) {
131131 return ;
132132 }
@@ -150,7 +150,7 @@ TEST(GemmRwriter, TransLeft) {
150150 CompareResult (&program, target, input_ids, {out->id }, 1 , passes, 123 , true );
151151}
152152
153- TEST (GemmRwriter , TransRight) {
153+ TEST (GemmRewriter , TransRight) {
154154 if (!cinn::runtime::IsCompiledWithCUDA ()) {
155155 return ;
156156 }
@@ -174,7 +174,7 @@ TEST(GemmRwriter, TransRight) {
174174 CompareResult (&program, target, input_ids, {out->id }, 1 , passes, 123 , true );
175175}
176176
177- TEST (GemmRwriter , TransTwo) {
177+ TEST (GemmRewriter , TransTwo) {
178178 if (!cinn::runtime::IsCompiledWithCUDA ()) {
179179 return ;
180180 }
@@ -199,7 +199,7 @@ TEST(GemmRwriter, TransTwo) {
199199 CompareResult (&program, target, input_ids, {out->id }, 2 , passes, 123 , true );
200200}
201201
202- TEST (GemmRwriter , NoTrans) {
202+ TEST (GemmRewriter , NoTrans) {
203203 if (!cinn::runtime::IsCompiledWithCUDA ()) {
204204 return ;
205205 }
@@ -222,7 +222,7 @@ TEST(GemmRwriter, NoTrans) {
222222 CompareResult (&program, target, input_ids, {out->id }, 0 , passes, 123 , true );
223223}
224224
225- TEST (GemmRwriter , BatchedComplex) {
225+ TEST (GemmRewriter , BatchedComplex) {
226226 if (!cinn::runtime::IsCompiledWithCUDA ()) {
227227 return ;
228228 }
@@ -256,7 +256,7 @@ TEST(GemmRwriter, BatchedComplex) {
256256 CompareResult (&program, target, input_ids, {out->id }, 4 , passes, 123 , false );
257257}
258258
259- TEST (GemmRwriter , Complex) {
259+ TEST (GemmRewriter , Complex) {
260260 if (!cinn::runtime::IsCompiledWithCUDA ()) {
261261 return ;
262262 }
0 commit comments