Edit

kc3-lang/angle/src/libANGLE/CLProgram.cpp

Branch :

  • Show log

    Commit

  • Author : John Plate
    Date : 2021-05-31 12:04:42
    Hash : daf18594
    Message : CL: Refactor front end entry functions Move all front end functions that are only called from the OpenCL entry points to a separate section at the top of the front end object class bodies, which improves readability. Bug: angleproject:6001 Change-Id: Id360adbf0c439e4068c379aa4ea1da25e99f7b53 Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/2928419 Reviewed-by: Jamie Madill <jmadill@chromium.org> Reviewed-by: Cody Northrop <cnorthrop@google.com>

  • src/libANGLE/CLProgram.cpp
  • //
    // Copyright 2021 The ANGLE Project Authors. All rights reserved.
    // Use of this source code is governed by a BSD-style license that can be
    // found in the LICENSE file.
    //
    // CLProgram.cpp: Implements the cl::Program class.
    
    #include "libANGLE/CLProgram.h"
    
    #include "libANGLE/CLContext.h"
    #include "libANGLE/CLPlatform.h"
    
    #include <cstring>
    
    namespace cl
    {
    
    cl_int Program::getInfo(ProgramInfo name, size_t valueSize, void *value, size_t *valueSizeRet) const
    {
        static_assert(std::is_same<cl_uint, cl_bool>::value &&
                          std::is_same<cl_uint, cl_addressing_mode>::value &&
                          std::is_same<cl_uint, cl_filter_mode>::value,
                      "OpenCL type mismatch");
    
        std::vector<cl_device_id> devices;
        std::vector<size_t> binarySizes;
        std::vector<const unsigned char *> binaries;
        cl_uint valUInt       = 0u;
        void *valPointer      = nullptr;
        const void *copyValue = nullptr;
        size_t copySize       = 0u;
    
        switch (name)
        {
            case ProgramInfo::ReferenceCount:
                valUInt   = getRefCount();
                copyValue = &valUInt;
                copySize  = sizeof(valUInt);
                break;
            case ProgramInfo::Context:
                valPointer = mContext->getNative();
                copyValue  = &valPointer;
                copySize   = sizeof(valPointer);
                break;
            case ProgramInfo::NumDevices:
                valUInt   = static_cast<decltype(valUInt)>(mDevices.size());
                copyValue = &valUInt;
                copySize  = sizeof(valUInt);
                break;
            case ProgramInfo::Devices:
                devices.reserve(mDevices.size());
                for (const DevicePtr &device : mDevices)
                {
                    devices.emplace_back(device->getNative());
                }
                copyValue = devices.data();
                copySize  = devices.size() * sizeof(decltype(devices)::value_type);
                break;
            case ProgramInfo::Source:
                copyValue = mSource.c_str();
                copySize  = mSource.length() + 1u;
                break;
            case ProgramInfo::IL:
                copyValue = mIL.c_str();
                copySize  = mIL.length() + 1u;
                break;
            case ProgramInfo::BinarySizes:
                binarySizes.resize(mDevices.size(), 0u);
                for (size_t index = 0u; index < binarySizes.size(); ++index)
                {
                    binarySizes[index] = index < mBinaries.size() ? mBinaries[index].size() : 0u;
                }
                copyValue = binarySizes.data();
                copySize  = binarySizes.size() * sizeof(decltype(binarySizes)::value_type);
                break;
            case ProgramInfo::Binaries:
                binaries.resize(mDevices.size(), nullptr);
                for (size_t index = 0u; index < binaries.size(); ++index)
                {
                    binaries[index] = index < mBinaries.size() && mBinaries[index].empty()
                                          ? mBinaries[index].data()
                                          : nullptr;
                }
                copyValue = binaries.data();
                copySize  = binaries.size() * sizeof(decltype(binaries)::value_type);
                break;
            case ProgramInfo::NumKernels:
                copyValue = &mNumKernels;
                copySize  = sizeof(mNumKernels);
                break;
            case ProgramInfo::KernelNames:
                copyValue = mKernelNames.c_str();
                copySize  = mKernelNames.length() + 1u;
                break;
            case ProgramInfo::ScopeGlobalCtorsPresent:
                valUInt   = CL_FALSE;
                copyValue = &valUInt;
                copySize  = sizeof(valUInt);
                break;
            case ProgramInfo::ScopeGlobalDtorsPresent:
                valUInt   = CL_FALSE;
                copyValue = &valUInt;
                copySize  = sizeof(valUInt);
                break;
            default:
                return CL_INVALID_VALUE;
        }
    
        if (value != nullptr)
        {
            // CL_INVALID_VALUE if size in bytes specified by param_value_size is < size of return type
            // as described in the Program Object Queries table and param_value is not NULL.
            if (valueSize < copySize)
            {
                return CL_INVALID_VALUE;
            }
            if (copyValue != nullptr)
            {
                std::memcpy(value, copyValue, copySize);
            }
        }
        if (valueSizeRet != nullptr)
        {
            *valueSizeRet = copySize;
        }
        return CL_SUCCESS;
    }
    
    cl_kernel Program::createKernel(const char *kernel_name, cl_int &errorCode)
    {
        return Object::Create<Kernel>(errorCode, *this, kernel_name);
    }
    
    cl_int Program::createKernels(cl_uint numKernels, cl_kernel *kernels, cl_uint *numKernelsRet)
    {
        if (kernels == nullptr)
        {
            numKernels = 0u;
        }
        rx::CLKernelImpl::CreateFuncs createFuncs;
        cl_int errorCode = mImpl->createKernels(numKernels, createFuncs, numKernelsRet);
        if (errorCode == CL_SUCCESS)
        {
            KernelPtrs krnls;
            krnls.reserve(createFuncs.size());
            while (!createFuncs.empty())
            {
                krnls.emplace_back(new Kernel(*this, createFuncs.front(), errorCode));
                if (errorCode != CL_SUCCESS)
                {
                    return CL_INVALID_VALUE;
                }
                createFuncs.pop_front();
            }
            for (KernelPtr &kernel : krnls)
            {
                *kernels++ = kernel.release();
            }
        }
        return errorCode;
    }
    
    Program::~Program() = default;
    
    Program::Program(Context &context, std::string &&source, cl_int &errorCode)
        : mContext(&context),
          mDevices(context.getDevices()),
          mImpl(context.getImpl().createProgramWithSource(*this, source, errorCode)),
          mSource(std::move(source))
    {}
    
    Program::Program(Context &context, const void *il, size_t length, cl_int &errorCode)
        : mContext(&context),
          mDevices(context.getDevices()),
          mIL(static_cast<const char *>(il), length),
          mImpl(context.getImpl().createProgramWithIL(*this, il, length, errorCode)),
          mSource(mImpl ? mImpl->getSource(errorCode) : std::string{})
    {}
    
    Program::Program(Context &context,
                     DevicePtrs &&devices,
                     Binaries &&binaries,
                     cl_int *binaryStatus,
                     cl_int &errorCode)
        : mContext(&context),
          mDevices(std::move(devices)),
          mImpl(context.getImpl().createProgramWithBinary(*this, binaries, binaryStatus, errorCode)),
          mSource(mImpl ? mImpl->getSource(errorCode) : std::string{}),
          mBinaries(std::move(binaries))
    {}
    
    Program::Program(Context &context, DevicePtrs &&devices, const char *kernelNames, cl_int &errorCode)
        : mContext(&context),
          mDevices(std::move(devices)),
          mImpl(context.getImpl().createProgramWithBuiltInKernels(*this, kernelNames, errorCode)),
          mSource(mImpl ? mImpl->getSource(errorCode) : std::string{})
    {}
    
    }  // namespace cl