Method configure

src/gpu/cl/operators/ClMatMul.cpp:88–149

}

void ClMatMul::configure(const CLCompileContext    &compile_context,
                         ITensorInfo               *lhs,
                         ITensorInfo               *rhs,
                         ITensorInfo               *dst,
                         const MatMulInfo          &matmul_info,
                         const ActivationLayerInfo &act_info)
{
    ARM_COMPUTE_ERROR_ON_NULLPTR(lhs, rhs, dst);
    ARM_COMPUTE_LOG_PARAMS(lhs, rhs, dst, matmul_info);

    // Perform validation step
    ARM_COMPUTE_ERROR_THROW_ON(validate(lhs, rhs, dst, matmul_info));

    const GPUTarget        gpu_target    = CLScheduler::get().target();
    const auto             kernel_config = ClMatMulNativeKernelConfigurationFactory::create(gpu_target);
    const MatMulKernelInfo kernel_info   = kernel_config->configure(lhs, rhs, matmul_info);

    const auto             kernel_selector = ClMatMulNativeKernelVariantFactory::create(gpu_target);
    const MatMulKernelType kernel_type     = kernel_selector->select_kernel(lhs, rhs, matmul_info, act_info);

    switch (kernel_type)
    {
        case MatMulKernelType::NATIVE_FP:
        {
            auto kernel = std::make_unique<ClMatMulNativeKernel>();
            kernel->set_target(gpu_target);

            kernel->configure(compile_context, lhs, rhs, nullptr /* bias */, dst, kernel_info, act_info);
            _matmul_kernel = std::move(kernel);
        }
        break;
        case MatMulKernelType::NATIVE_MMUL_FP:
        {
            auto kernel = std::make_unique<ClMatMulNativeMMULKernel>();
            kernel->set_target(gpu_target);

            kernel->configure(compile_context, lhs, rhs, nullptr /* bias */, dst, kernel_info);
            _matmul_kernel = std::move(kernel);
        }
        break;
        case MatMulKernelType::NATIVE_QUANTIZED:
        {
            auto kernel = std::make_unique<ClMatMulLowpNativeKernel>();
            kernel->set_target(gpu_target);

            kernel->configure(compile_context, lhs, rhs, nullptr /* bias */, dst, kernel_info, act_info);
            _matmul_kernel = std::move(kernel);
        }
        break;
        case MatMulKernelType::NATIVE_MMUL_QUANTIZED:
        {
            auto kernel = std::make_unique<ClMatMulLowpNativeMMULKernel>();
            kernel->set_target(gpu_target);

            kernel->configure(compile_context, lhs, rhs, nullptr /* bias */, dst, kernel_info, act_info);
            _matmul_kernel = std::move(kernel);
        }
        break;