Skip to content

Commit a4b6769

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 5b86955 commit a4b6769

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

  • transformer_engine/jax/csrc/extensions

transformer_engine/jax/csrc/extensions/gemm.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -756,11 +756,10 @@ Error_Type GroupedGemmFFI(cudaStream_t stream, Buffer_Type lhs_data, Buffer_Type
756756
nullptr, // config (use defaults)
757757
stream);
758758

759-
760759
fprintf(stderr, "After GEMM:\n");
761760
NVTE_CHECK_CUDA(cudaStreamSynchronize(stream));
762761
// fflush(stderr);
763-
762+
764763
return ffi_with_cuda_error_check();
765764
}
766765

0 commit comments

Comments
 (0)