// Note that cublas follows fortran order.
int lda = (TransA == CblasNoTrans) ? K : M;
int ldb = (TransB == CblasNoTrans) ? N : K;
- cublasOperation_t cuTransA =
+ cublasOperation_t cuTransA =
(TransA == CblasNoTrans) ? CUBLAS_OP_N : CUBLAS_OP_T;
cublasOperation_t cuTransB =
(TransB == CblasNoTrans) ? CUBLAS_OP_N : CUBLAS_OP_T;
CUBLAS_CHECK(cublasSgemm(Caffe::cublas_handle(), cuTransB, cuTransA,
- N, M, K, &alpha, B, ldb, A, lda, &beta, C, N));
+ N, M, K, &alpha, B, ldb, A, lda, &beta, C, N));
}
template <>
// Note that cublas follows fortran order.
int lda = (TransA == CblasNoTrans) ? K : M;
int ldb = (TransB == CblasNoTrans) ? N : K;
- cublasOperation_t cuTransA =
+ cublasOperation_t cuTransA =
(TransA == CblasNoTrans) ? CUBLAS_OP_N : CUBLAS_OP_T;
cublasOperation_t cuTransB =
(TransB == CblasNoTrans) ? CUBLAS_OP_N : CUBLAS_OP_T;
CUBLAS_CHECK(cublasDgemm(Caffe::cublas_handle(), cuTransB, cuTransA,
- N, M, K, &alpha, B, ldb, A, lda, &beta, C, N));
+ N, M, K, &alpha, B, ldb, A, lda, &beta, C, N));
}
template <>
}
template <>
-void caffe_sqr<float>(const int n, const float* a, float* y){
+void caffe_sqr<float>(const int n, const float* a, float* y) {
vsSqr(n, a, y);
}
template <>
void caffe_vRngGaussian<float>(const int n, float* r, const float a,
- const float sigma){
+ const float sigma) {
VSL_CHECK(vsRngGaussian(VSL_RNG_METHOD_GAUSSIAN_BOXMULLER,
Caffe::vsl_stream(), n, r, a, sigma));
}
template <>
void caffe_vRngGaussian<double>(const int n, double* r, const double a,
- const double sigma){
+ const double sigma) {
VSL_CHECK(vdRngGaussian(VSL_RNG_METHOD_GAUSSIAN_BOXMULLER,
Caffe::vsl_stream(), n, r, a, sigma));
}