diff --git a/mlx/backend/cuda/gemms/cublas_gemm.cpp b/mlx/backend/cuda/gemms/cublas_gemm.cpp index 65cf3beb3..a0e936fd4 100644 --- a/mlx/backend/cuda/gemms/cublas_gemm.cpp +++ b/mlx/backend/cuda/gemms/cublas_gemm.cpp @@ -7,14 +7,6 @@ #include -void check_cublas_error(const char* name, cublasStatus_t err) { - if (err != CUBLAS_STATUS_SUCCESS) { - // TODO: Use cublasGetStatusString when it is widely available. - throw std::runtime_error( - fmt::format("{} failed with code: {}.", name, static_cast(err))); - } -} - namespace mlx::core::cu { struct CublasPreference { diff --git a/mlx/backend/cuda/gemms/cublas_gemm.h b/mlx/backend/cuda/gemms/cublas_gemm.h index 2babb808e..eccee8580 100644 --- a/mlx/backend/cuda/gemms/cublas_gemm.h +++ b/mlx/backend/cuda/gemms/cublas_gemm.h @@ -7,10 +7,6 @@ #include #include -void check_cublas_error(const char* name, cublasStatus_t err); - -#define CHECK_CUBLAS_ERROR(cmd) check_cublas_error(#cmd, (cmd)) - namespace mlx::core::cu { class Matmul { public: