mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 10:14:27 +00:00
Remove cublas dependency
This commit is contained in:
@@ -1,4 +1,3 @@
|
||||
#include <cublas_v2.h>
|
||||
#include <cuda_runtime.h>
|
||||
|
||||
#include <cstdio>
|
||||
@@ -6,7 +5,7 @@
|
||||
|
||||
#include "cuda_helper.cuh"
|
||||
|
||||
cudaDeviceProp initializeCUDA(cublasHandle_t& cublasHandle) {
|
||||
cudaDeviceProp initializeCUDA() {
|
||||
int deviceCount;
|
||||
CUDA_CHECK(cudaGetDeviceCount(&deviceCount));
|
||||
|
||||
@@ -23,8 +22,5 @@ cudaDeviceProp initializeCUDA(cublasHandle_t& cublasHandle) {
|
||||
|
||||
std::printf("Using CUDA device %d: %s\n", device, deviceProp.name);
|
||||
|
||||
// Initialize cuBLAS
|
||||
CUBLAS_CHECK(cublasCreate(&cublasHandle));
|
||||
|
||||
return deviceProp;
|
||||
}
|
||||
Reference in New Issue
Block a user