zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

comment

Browse Source
This commit is contained in:
Awni Hannun
2025-07-24 19:11:06 -07:00
parent e1d40be0b6
commit dfb76fc8f2
2 changed files with 0 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/backend/cuda/gemms/cublas_gemm.cpp
View File

@@ -7,14 +7,6 @@
#include <fmt/format.h> #include <fmt/format.h>
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<int>(err)));
}
}
namespace mlx::core::cu { namespace mlx::core::cu {
struct CublasPreference { struct CublasPreference {

4
mlx/backend/cuda/gemms/cublas_gemm.h
View File

@@ -7,10 +7,6 @@
#include <cublasLt.h> #include <cublasLt.h>
#include <optional> #include <optional>
void check_cublas_error(const char* name, cublasStatus_t err);
#define CHECK_CUBLAS_ERROR(cmd) check_cublas_error(#cmd, (cmd))
namespace mlx::core::cu { namespace mlx::core::cu {
class Matmul { class Matmul {
public: public: