Method assignKargs

src/library/tools/ktest/step.cpp:496–616

}

void
Step::assignKargs(const StepKargs& map)
{
    CLBlasKargs args;
    KernelArg kargsList[MAX_KERNEL_ARGS];
    Variable *v;

    if ((pattern_->sops == NULL) || (pattern_->sops->assignKargs == NULL)) {
        return;
    }

    memset(&kargsList, KARG_NONE, sizeof(kargsList));

    args.kernType = CLBLAS_COMPUTING_KERNEL;
    args.dtype = TYPE_COMPLEX_DOUBLE;
    args.addrBits = 0;

    args.order = static_cast<clblasOrder>(KARG_ORDER);
    args.side = static_cast<clblasSide>(KARG_SIDE);
    args.uplo = static_cast<clblasUplo>(KARG_UPLO);
    args.transA = static_cast<clblasTranspose>(KARG_TRANS_A);
    args.transB = static_cast<clblasTranspose>(KARG_TRANS_B);
    args.diag = static_cast<clblasDiag>(KARG_DIAG);

    args.M = KARG_M;
    args.N = KARG_N;
    args.K = KARG_K;

    args.lda.matrix = KARG_LDA;
    args.ldb.matrix = KARG_LDB;
    args.ldc.matrix = KARG_LDC;

    args.offsetM = KARG_OFFSET_M;
    args.offsetN = KARG_OFFSET_N;
    args.offsetK = KARG_OFFSET_K;
    args.offA = KARG_OFF_A;
    args.offBX = KARG_OFF_BX;
    args.offCY = KARG_OFF_CY;

    args.A = reinterpret_cast<cl_mem>(KARG_A);
    args.B = reinterpret_cast<cl_mem>(KARG_B);
    args.C = reinterpret_cast<cl_mem>(KARG_C);

    memset(&args.alpha, KARG_ALPHA, sizeof(args.alpha));
    memset(&args.beta, KARG_BETA, sizeof(args.beta));

    args.scimage[0] = reinterpret_cast<cl_mem>(KARG_SCIMAGE_0);
    args.scimage[1] = reinterpret_cast<cl_mem>(KARG_SCIMAGE_1);

    pattern_->sops->assignKargs(kargsList, static_cast<void*>(&args), &kextra_);

    for (unsigned int i = 0; (i < MAX_KERNEL_ARGS) && (kargsList[i].typeSize != 0); i++) {
        switch (static_cast<StepKarg>(kargsList[i].arg.data[0])) {
        case KARG_M:
            v = map.M;
            break;
        case KARG_N:
            v = map.N;