@specs-feup/clava/api/clava/opencl/OpenCLCall.js

A C/C++ source-to-source compiler written in Typescript

import Io from "@specs-feup/lara/api/lara/Io.js";
import Platforms from "@specs-feup/lara/api/lara/Platforms.js";
import IdGenerator from "@specs-feup/lara/api/lara/util/IdGenerator.js";
import OpenCLCallVariables from "./OpenCLCallVariables.js";
export default class OpenCLCall {
    $kernel = undefined;
    deviceId = 1;
    setKernel($function) {
        if (!$function.getAncestor("file").isOpenCL) {
            throw new Error("OpenCLCall.setKernel: expected a function in an OpenCL file");
        }
        this.$kernel = $function;
    }
    setDeviceId(deviceId) {
        this.deviceId = deviceId;
    }
    replaceCall($call) {
        this.replaceCallPreconditions();
        // Add include
        this.addOpenCLInclude($call);
        // Generate id
        const id = IdGenerator.next("opencl_call_");
        const variables = new OpenCLCallVariables(id);
        this.loadKernelFile($call, variables);
        this.clInit($call, variables);
    }
    replaceCallPreconditions() {
        if (this.$kernel == undefined) {
            throw new Error("OpenCLCall._replaceCallPreconditions: Expected kernel to be set");
        }
    }
    addOpenCLInclude($call) {
        const $file = $call.getAncestor("file");
        // If MacOS, add include <OpenCL/opencl.h>
        if (Platforms.isMac()) {
            $file.addInclude("OpenCL/opencl.h", true);
            return;
        }
        // Otherwise, <CL/cl.h>
        $file.addInclude("CL/cl.h", true);
    }
    loadKernelFile($call, variables) {
        // Get necessary data
        const $kernelFile = $call.getAncestor("file");
        const kernelPath = $kernelFile.relativeFilepath;
        const kernelFileBytes = Io.getPath($kernelFile.filepath).length;
        // Insert before the call
        $call.insertBefore(`// Load the kernel source code into the array
FILE *${variables.getKernelFile()} = fopen("${kernelPath}", "r");
if (!${variables.getKernelFile()}) {
	fprintf(stdout, "Failed to load kernel.\n");
	exit(1);
}
char *${variables.getKernelString()} = (char*)malloc(${kernelFileBytes});
size_t ${variables.getKernelStringSize()} = fread(${variables.getKernelString()}, 1, ${kernelFileBytes}, ${variables.getKernelFile()});
fclose(${variables.getKernelFile()});`);
    }
    /**
     * This only needs to be done once per function
     */
    clInit($call, variables) {
        // TODO: Set of functions where this has been called
        // Insert before the call
        $call.insertBefore(`cl_int ${variables.getErrorCode()};
cl_uint ${variables.getNumPlatforms()};
cl_platform_id ${variables.getPlatformId()};

// Check the number of platforms
${variables.getErrorCode()} = clGetPlatformIDs(0, NULL, &${variables.getNumPlatforms()});
if(${variables.getErrorCode()} != CL_SUCCESS) {
	fprintf(stderr, "[OpenCL] Error getting number of platforms\n");
	exit(1);
} else if(${variables.getNumPlatforms()} == 0) {
	fprintf(stderr, "[OpenCL] No platforms found.\n");
	exit(1);
} else {
	printf("[OpenCL] Number of platforms is %d\n",${variables.getNumPlatforms()});
}

${variables.getErrorCode()} = clGetPlatformIDs(${this.deviceId}, &${variables.getPlatformId()}, NULL);
if(${variables.getErrorCode()} != CL_SUCCESS) {
	fprintf(stderr, "[OpenCL] Error getting platform ID for device ${this.deviceId}.\n");
	exit(1);
}`);
    }
}
//# sourceMappingURL=OpenCLCall.js.map