Function prepareSerial

numexpr3/interpreter.cpp:106–139

This is the equivalent of prepareThreads but for serial mode, where the original self and iter aren't copied.

// This is the equivalent of prepareThreads but for serial mode, where the original
// self and iter aren't copied.
int 
prepareSerial( NumExprObject* self, NpyIter *iter, int *pc_error, char **errorMessage ) {
    int I = 0, R = 0; 
    npy_intp memOffset = 0, taskFactor, numBlocks;

    // Setup error tracking
    gs.ret_code = 0;
    gs.pc_error = pc_error;
    gs.errorMessage = errorMessage;

    // printf( "prepareSerial()\n");

    NpyIter_GetIterIndexRange(iter, &gs.start, &gs.vlen);

    taskFactor = TASKS_PER_THREAD*BLOCK_SIZE1*gs.n_thread;
    numBlocks = (gs.vlen - gs.start + taskFactor - 1) /
                            taskFactor;
    gs.task_size = numBlocks * BLOCK_SIZE1; // Note this is much bigger than BLOCK_SIZE1

    if( BLOCK_SIZE1 * self->total_temp_itemsize >  gs.tempSize ) {
        numexpr_set_tempsize( BLOCK_SIZE1 * self->total_temp_itemsize );
        // printf("Engorged: gs.tempSize = %d, gs.tempArena = %p\n", gs.tempSize, gs.tempArena );
    }
    // Setup temporory memory pointers
    for( R=0; R < self->n_reg; R++ ) {
        if( self->registers[R].kind != KIND_TEMP ) continue;
        
        // printf( "reg#%d has kind %d and itemsize %d\n", R, self->registers[R].kind, (int)self->registers[R].itemsize );
        self->registers[R].mem = gs.tempArena + memOffset;
        // printf( "    thread #0, reg #%d, points to %p\n", R, self->registers[R].mem );
        memOffset += BLOCK_SIZE1 * self->registers[R].itemsize;
    }
    return 0;
}

// This function prepares the pointers towards the temporary memory block, and 
// the NumExprObjects used by threads.