1 /* 2 * Copyright (c) 2024, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 #include "opencl_backend.h" 26 27 28 /* 29 OpenCLKernel 30 */ 31 32 OpenCLBackend::OpenCLProgram::OpenCLKernel::OpenCLKernel(Backend::CompilationUnit *compilationUnit, char* name, cl_kernel kernel) 33 : Backend::CompilationUnit::Kernel(compilationUnit, name), kernel(kernel){ 34 } 35 36 OpenCLBackend::OpenCLProgram::OpenCLKernel::~OpenCLKernel() { 37 clReleaseKernel(kernel); 38 } 39 40 41 /* 42 void dispatchKernel(Kernel kernel, KernelContext kc, Arg ... args) { 43 for (int argn = 0; argn<args.length; argn++){ 44 Arg arg = args[argn]; 45 if (alwaysCopyBuffers || (((arg.flags &JavaDirty)==JavaDirty) && kernel.readsFrom(arg))) { 46 enqueueCopyToDevice(arg); 47 } 48 } 49 enqueueKernel(kernel); 50 waitForKernel(); 51 52 for (int argn = 0; argn<args.length; argn++){ 53 Arg arg = args[argn]; 54 if (alwaysCopyBuffers){ 55 enqueueCopyFromDevice(arg); 56 arg.flags = 0; 57 }else{ 58 if (kernel.writesTo(arg)) { 59 arg.flags = DeviceDirty; 60 }else{ 61 arg.flags = 0; 62 } 63 } 64 } 65 66 } 67 */ 68 69 bool OpenCLBackend::OpenCLProgram::OpenCLKernel::setArg(KernelArg *arg, Buffer *buffer){ 70 auto * openCLBuffer = dynamic_cast<OpenCLBuffer *>(buffer); 71 cl_int status = clSetKernelArg(kernel, arg->idx, sizeof(cl_mem), &openCLBuffer->clMem); 72 if (status != CL_SUCCESS) { 73 std::cerr << OpenCLBackend::errorMsg(status) << std::endl; 74 return false; 75 } 76 return true; 77 } 78 bool OpenCLBackend::OpenCLProgram::OpenCLKernel::setArg(KernelArg *arg) { 79 cl_int status = clSetKernelArg(kernel, arg->idx, arg->size(), (void *) &arg->value); 80 if (status != CL_SUCCESS) { 81 std::cerr << OpenCLBackend::errorMsg(status) << std::endl; 82 return false; 83 } 84 return true; 85 }