Skip to content
Extraits de code Groupes Projets
Valider f7e9ca70 rédigé par JordanHanotiaux's avatar JordanHanotiaux
Parcourir les fichiers

Update matrix_opencl.cpp

parent 83682376
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
......@@ -192,7 +192,7 @@ void KernelCache::compileKernels(cl::Context context, const std::vector<cl::Devi
kernel_matrix_mul = cl::Kernel(prog_matrix_mul, "matrix_mul");
cl::Program prog_matrix_mul_V2 = loadAndBuildProgram(context, devices, kernel_source_matrix_mul_V2, "matrix_mul_V2");
kernel_matrix_mul_V2 = cl::Kernel(prog_matrix_mul_V2, "matrix_mul_V2");
kernel_matrix_mul_v2 = cl::Kernel(prog_matrix_mul_V2, "matrix_mul_V2");
cl::Program prog_sigmoid = loadAndBuildProgram(context, devices, kernel_source_sigmoid, "sigmoid");
kernel_sigmoid = cl::Kernel(prog_sigmoid, "sigmoid");
......@@ -380,7 +380,7 @@ MatrixCL MatrixCL::operator*(const MatrixCL& other) const {
MatrixCL MatrixCL::matrix_mul_V2(const MatrixCL& other) const {
MatrixCL result(rows_, other.numCols(), context_, queue_);
cl::Kernel kernel = kernels_->kernel_matrix_mul_V2;
cl::Kernel kernel = kernels_->kernel_matrix_mul_v2;
kernel.setArg(0, buffer_);
kernel.setArg(1, other.getBuffer());
kernel.setArg(2, result.getBuffer());
......
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter