Esempio n. 1
0
  CLProgram(CLContext context, Map<CLDevice, byte[]> binaries, String source) {
    super(null, true);
    this.context = context;
    this.source = source;

    setBinaries(binaries);
  }
Esempio n. 2
0
  /** Returns the context of this program */
  public synchronized CLProgram build() throws CLBuildException {
    if (built) throw new IllegalThreadStateException("Program was already built !");

    String contentSignature = null;
    File cacheFile = null;
    boolean readBinaries = false;
    if (isCached()) {
      try {
        contentSignature = computeCacheSignature();
        byte[] sha =
            java.security.MessageDigest.getInstance("MD5")
                .digest(contentSignature.getBytes(textEncoding));
        StringBuilder shab = new StringBuilder();
        for (byte b : sha) shab.append(Integer.toHexString(b & 0xff));
        String hash = shab.toString();
        cacheFile = new File(JavaCL.userCacheDir, hash);
        if (cacheFile.exists()) {
          Pair<Map<CLDevice, byte[]>, String> bins =
              readBinaries(
                  Arrays.asList(getDevices()), contentSignature, new FileInputStream(cacheFile));
          setBinaries(bins.getFirst());
          this.source = bins.getSecond();
          assert log(Level.INFO, "Read binaries cache from '" + cacheFile + "'");
          readBinaries = true;
        }
      } catch (Exception ex) {
        assert log(Level.WARNING, "Failed to load cached program", ex);
        entity = null;
      }
    }

    if (entity == null) allocate();

    Runnable deleteTempFiles = null;
    if (!readBinaries)
      try {
        deleteTempFiles = copyIncludesToTemporaryDirectory();
      } catch (IOException ex) {
        throw new CLBuildException(this, ex.toString(), Collections.EMPTY_LIST);
      }

    int nDevices = devices.length;
    cl_device_id[] deviceIds = null;
    if (nDevices != 0) {
      deviceIds = new cl_device_id[nDevices];
      for (int i = 0; i < nDevices; i++) deviceIds[i] = devices[i].getEntity();
    }
    int err =
        CL.clBuildProgram(
            getEntity(), nDevices, deviceIds, readBinaries ? null : getOptionsString(), null, null);
    // int err = CL.clBuildProgram(getEntity(), 0, null, getOptionsString(), null, null);
    if (err != CL_SUCCESS) { // BUILD_PROGRAM_FAILURE) {
      NativeSizeByReference len = new NativeSizeByReference();
      int bufLen = 2048 * 32; // TODO find proper size
      Memory buffer = new Memory(bufLen);

      HashSet<String> errs = new HashSet<String>();
      if (deviceIds == null) {
        error(
            CL.clGetProgramBuildInfo(
                getEntity(), null, CL_PROGRAM_BUILD_LOG, toNS(bufLen), buffer, len));
        String s = buffer.getString(0);
        errs.add(s);
      } else
        for (cl_device_id device : deviceIds) {
          error(
              CL.clGetProgramBuildInfo(
                  getEntity(), device, CL_PROGRAM_BUILD_LOG, toNS(bufLen), buffer, len));
          String s = buffer.getString(0);
          errs.add(s);
        }

      throw new CLBuildException(this, "Compilation failure : " + errorString(err), errs);
    }
    built = true;
    if (deleteTempFiles != null) deleteTempFiles.run();

    if (isCached() && !readBinaries) {
      JavaCL.userCacheDir.mkdirs();
      try {
        Map<CLDevice, byte[]> binaries = getBinaries();
        if (!binaries.isEmpty()) {
          writeBinaries(
              getBinaries(), getSource(), contentSignature, new FileOutputStream(cacheFile));
          assert log(Level.INFO, "Wrote binaries cache to '" + cacheFile + "'");
        }
      } catch (Exception ex) {
        new IOException("[JavaCL] Failed to cache program", ex).printStackTrace();
      }
    }

    return this;
  }