Skip to content

GitLab

Commit e6334634 authored by Bartlomiej Kocot's avatar Bartlomiej Kocot Committed by Bartłomiej Kocot
Browse files

Fix batched_gemm tests

parent 91c33aec
Show whitespace changes
Inline Side-by-side
Showing with 48 additions and 48 deletions
test/batched_gemm/batched_gemm_bf16.cpp
View file @ e6334634
...@@ -38,9 +38,9 @@ int main() ...@@ -38,9 +38,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -58,9 +58,9 @@ int main() ...@@ -58,9 +58,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -78,9 +78,9 @@ int main() ...@@ -78,9 +78,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -98,9 +98,9 @@ int main() ...@@ -98,9 +98,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
......
test/batched_gemm/batched_gemm_fp16.cpp
View file @ e6334634
...@@ -38,9 +38,9 @@ int main() ...@@ -38,9 +38,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -58,9 +58,9 @@ int main() ...@@ -58,9 +58,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -78,9 +78,9 @@ int main() ...@@ -78,9 +78,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -98,9 +98,9 @@ int main() ...@@ -98,9 +98,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
......
test/batched_gemm/batched_gemm_fp32.cpp
View file @ e6334634
...@@ -38,9 +38,9 @@ int main() ...@@ -38,9 +38,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -58,9 +58,9 @@ int main() ...@@ -58,9 +58,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -78,9 +78,9 @@ int main() ...@@ -78,9 +78,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -98,9 +98,9 @@ int main() ...@@ -98,9 +98,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
......
test/batched_gemm/batched_gemm_int8.cpp
View file @ e6334634
...@@ -38,9 +38,9 @@ int main() ...@@ -38,9 +38,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -58,9 +58,9 @@ int main() ...@@ -58,9 +58,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Row,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -78,9 +78,9 @@ int main() ...@@ -78,9 +78,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Row,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
...@@ -98,9 +98,9 @@ int main() ...@@ -98,9 +98,9 @@ int main()
PassThrough, PassThrough,
PassThrough, PassThrough,
PassThrough, PassThrough,
DeviceBatchedGemm<ALayout, DeviceBatchedGemm<Col,
BLayout, Col,
CLayout, Row,
ADataType, ADataType,
BDataType, BDataType,
CDataType, CDataType,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment