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 }