Skip to content

Commit 4014ae3

Browse files
committed
Add test case for k_bin_bcast_unravel in CUDA backend
1 parent 3cd6708 commit 4014ae3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/test-backend-ops.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6039,6 +6039,9 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
60396039
add_test_bin_bcast(type, {10, 5, 4, 3}, {1, 2, 2, 2});
60406040
add_test_bin_bcast(type, {10, 5, 4, 3}, {2, 2, 2, 2});
60416041

6042+
// test case for k_bin_bcast_unravel in CUDA backend
6043+
add_test_bin_bcast(type, {1, 1, 65536, 1}, {256, 1, 1, 1});
6044+
60426045
// stable diffusion
60436046
add_test_bin_bcast(type, {1280, 1, 1, 1}, {1, 1, 1, 1});
60446047
add_test_bin_bcast(type, {1280, 1, 1, 1}, {1, 16, 16, 1});

0 commit comments

Comments
 (0)