Skip to content

Commit 1545982

Browse files
committed
Rebase to use compat API
1 parent d29944f commit 1545982

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

examples/11_bmg_moe_gemm_bf16/11_bmg_moe_gemm_bf16.cpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ template <class Gemm> struct ExampleRunner {
274274
);
275275

276276
// Wait for kernel to finish
277-
syclcompat::wait();
277+
compat::wait();
278278

279279
// Check if output from CUTLASS kernel and reference kernel are equal or
280280
// not
@@ -509,14 +509,14 @@ template <class Gemm> struct ExampleRunner {
509509
// Run the GEMM
510510
CUTLASS_CHECK(gemm_op.run());
511511

512-
syclcompat::wait();
512+
compat::wait();
513513
initialize(options);
514514
// Verify that the result is correct
515515
bool passed = verify(options);
516516
std::cout << "Disposition: " << (passed ? "Passed" : "Failed") << std::endl;
517517
if (!passed)
518518
return cutlass::Status::kErrorInternal;
519-
syclcompat::wait();
519+
compat::wait();
520520
arguments = args_from_options(options, hw_info, A_ptr, B_ptr, D_ptr, gemm_n, gemm_k);
521521
CUTLASS_CHECK(gemm_op.can_implement(arguments));
522522

@@ -528,7 +528,7 @@ template <class Gemm> struct ExampleRunner {
528528
for (int iter = 0; iter < options.iterations; ++iter) {
529529
CUTLASS_CHECK(gemm_op.run());
530530
}
531-
syclcompat::wait();
531+
compat::wait();
532532

533533
float cute_time = timer.seconds() * 1000;
534534
double cute_average_time = double(cute_time) / double(options.iterations);
@@ -718,3 +718,4 @@ int main(int argc, const char **argv) {
718718

719719
return 0;
720720
}
721+

0 commit comments

Comments
 (0)