OpenCL-CTS/test_conformance/half/main.c

//
// Copyright (c) 2017 The Khronos Group Inc.
// 
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#if !defined (_WIN32)
#include <sys/resource.h>
#if !defined(__ANDROID__)
#include <sys/sysctl.h>
#endif
#include <libgen.h>
#include <sys/param.h>
#endif

#include "../../test_common/harness/testHarness.h"
#include "../../test_common/harness/mingw_compat.h"
#include "../../test_common/harness/parseParameters.h"
#if defined (__MINGW32__)
#include <sys/param.h>
#endif

#include "cl_utils.h"
#include "tests.h"

const char **   argList = NULL;
size_t          argCount = 0;
char            appName[64] = "ctest";
const char *addressSpaceNames[] = {"global", "private", "local", "constant"};

#pragma mark -
#pragma mark Declarations


static int ParseArgs( int argc, const char **argv );
static void PrintUsage( void );
static void PrintArch(void);
static void PrintDevice(void);


int g_arrVecSizes[kVectorSizeCount+kStrangeVectorSizeCount];
int g_arrVecAligns[kLargestVectorSize+1];
static int arrStrangeVecSizes[kStrangeVectorSizeCount] = {3};

basefn basefn_list[] = {
    Test_vload_half,
    Test_vloada_half,
    Test_vstore_half,
    Test_vstorea_half,
    Test_vstore_half_rte,
    Test_vstorea_half_rte,
    Test_vstore_half_rtz,
    Test_vstorea_half_rtz,
    Test_vstore_half_rtp,
    Test_vstorea_half_rtp,
    Test_vstore_half_rtn,
    Test_vstorea_half_rtn,
    Test_roundTrip,
};

const char *basefn_names[] = {
    "vload_half",
    "vloada_half",
    "vstore_half",
    "vstorea_half",
    "vstore_half_rte",
    "vstorea_half_rte",
    "vstore_half_rtz",
    "vstorea_half_rtz",
    "vstore_half_rtp",
    "vstorea_half_rtp",
    "vstore_half_rtn",
    "vstorea_half_rtn",
    "roundTrip",
};

ct_assert((sizeof(basefn_names) / sizeof(basefn_names[0])) == (sizeof(basefn_list) / sizeof(basefn_list[0])));

int num_fns = sizeof(basefn_names) / sizeof(char *);

int main (int argc, const char **argv )
{
    int error;
    int i;
    int alignbound;

    for(i = 0; i < kVectorSizeCount; ++i) {
      g_arrVecSizes[i] = (1<<i);
    }
    for(i = 0; i < kStrangeVectorSizeCount; ++i) {
      g_arrVecSizes[i+kVectorSizeCount] =
    arrStrangeVecSizes[i];
    }

    for(i = 0, alignbound=1; i <= kLargestVectorSize; ++i) {
    while(alignbound < i) {
        alignbound = alignbound<<1;
    }
    g_arrVecAligns[i] = alignbound;
    }

    test_start();

    argc = parseCustomParam(argc, argv);
    if (argc == -1)
    {
        test_finish();
        return -1;
    }

    if( (error = ParseArgs( argc, argv )) )
        goto exit;

    if( (error = InitCL()) )
        goto exit;

    if (gIsEmbedded) {
        vlog( "\tProfile: Embedded\n" );
    }else
    {
        vlog( "\tProfile: Full\n" );
    }

    fflush( stdout );
    error = parseAndCallCommandLineTests( argCount, argList, NULL, num_fns, basefn_list, basefn_names, true, 0, 0 );

exit:
    if(gQueue)
    {
        int flush_error = clFinish(gQueue);
        if(flush_error)
        {
            vlog_error("clFinish failed: %d\n", flush_error);
        }
    }

    if(gFailCount > 0)
    {
        vlog_error("FAILED %d sub-tests.\n", gFailCount);
    }

    ReleaseCL();
    test_finish();

    return error;
}

#pragma mark -
#pragma mark setup

static int ParseArgs( int argc, const char **argv )
{
    int i;
    argList = (const char **)calloc( argc - 1, sizeof( char*) );

    if( NULL == argList )
    {
        vlog_error( "Failed to allocate memory for argList.\n" );
        return 1;
    }

    argList[0] = argv[0];
    argCount = 1;

#if (defined( __APPLE__ ) || defined(__linux__) || defined(__MINGW32__))
    { // Extract the app name
        char baseName[ MAXPATHLEN ];
        strncpy( baseName, argv[0], MAXPATHLEN );
        char *base = basename( baseName );
        if( NULL != base )
        {
            strncpy( appName, base, sizeof( appName )  );
            appName[ sizeof( appName ) -1 ] = '\0';
        }
    }
#elif defined (_WIN32)
    {
        char fname[_MAX_FNAME + _MAX_EXT + 1];
        char ext[_MAX_EXT];

        errno_t err = _splitpath_s( argv[0], NULL, 0, NULL, 0,
                                   fname, _MAX_FNAME, ext, _MAX_EXT );
        if (err == 0) { // no error
            strcat (fname, ext); //just cat them, size of frame can keep both
            strncpy (appName, fname, sizeof(appName));
            appName[ sizeof( appName ) -1 ] = '\0';
        }
    }
#endif

    /* Check for environment variable to set device type */
    char *env_mode = getenv( "CL_DEVICE_TYPE" );
    if( env_mode != NULL )
    {
        if( strcmp( env_mode, "gpu" ) == 0 || strcmp( env_mode, "CL_DEVICE_TYPE_GPU" ) == 0 )
            gDeviceType = CL_DEVICE_TYPE_GPU;
        else if( strcmp( env_mode, "cpu" ) == 0 || strcmp( env_mode, "CL_DEVICE_TYPE_CPU" ) == 0 )
            gDeviceType = CL_DEVICE_TYPE_CPU;
        else if( strcmp( env_mode, "accelerator" ) == 0 || strcmp( env_mode, "CL_DEVICE_TYPE_ACCELERATOR" ) == 0 )
            gDeviceType = CL_DEVICE_TYPE_ACCELERATOR;
        else if( strcmp( env_mode, "default" ) == 0 || strcmp( env_mode, "CL_DEVICE_TYPE_DEFAULT" ) == 0 )
            gDeviceType = CL_DEVICE_TYPE_DEFAULT;
        else
        {
            vlog_error( "Unknown CL_DEVICE_TYPE env variable setting: %s.\nAborting...\n", env_mode );
            abort();
        }
    }

    unsigned int num_devices;
    cl_platform_id platform = NULL;
    clGetPlatformIDs(1, &platform, NULL);
    clGetDeviceIDs(platform, gDeviceType, 0, NULL, &num_devices);

    const char* device_index_env = getenv("CL_DEVICE_INDEX");
    if (device_index_env) {
        if (device_index_env) {
            gDeviceIndex = atoi(device_index_env);
        }

        if (gDeviceIndex >= num_devices) {
            vlog("Specified CL_DEVICE_INDEX=%d out of range, using index 0.\n",
                gDeviceIndex);
            gDeviceIndex = 0;
        }
    }

    vlog( "\n%s", appName );
    for( i = 1; i < argc; i++ )
    {
        const char *arg = argv[i];
        if( NULL == arg )
            break;

        vlog( "\t%s", arg );
        if( arg[0] == '-' )
        {
            arg++;
            while( *arg != '\0' )
            {
                switch( *arg )
                {
                    case 'd':
                        gTestDouble ^= 1;
                        break;

                    case 'h':
                        PrintUsage();
                        return -1;

                    case 't':
                        gReportTimes ^= 1;
                        break;

                    case 'w':  // Wimpy mode
                        gWimpyMode = true;
                        break;
                    case '[':
                        // wimpy reduction factor can be set with the option -[2^n]
                        // Default factor is 512, and n practically can be from 1 to 12
                        {
                            const char *arg_temp = strchr(&arg[1], ']');
                            if (arg_temp != 0)
                            {
                                int new_factor = atoi(&arg[1]);
                                arg = arg_temp; // Advance until ']'
                                if (new_factor && !(new_factor & (new_factor - 1)))
                                {
                                    vlog(" WimpyReduction factor changed from %d to %d \n", gWimpyReductionFactor, new_factor);
                                    gWimpyReductionFactor = new_factor;
                                }
                               else
                                {
                                   vlog(" Error in WimpyReduction factor, must be power of 2 \n");
                                }
                            }
                        }
                        break;
                    default:
                        vlog_error( " <-- unknown flag: %c (0x%2.2x)\n)", *arg, *arg );
                        PrintUsage();
                        return -1;
                }
                arg++;
            }
        }
        else
        {
            if( 0 == strcmp( arg, "CL_DEVICE_TYPE_CPU" ) )
                gDeviceType = CL_DEVICE_TYPE_CPU;
            else if( 0 == strcmp( arg, "CL_DEVICE_TYPE_GPU" ) )
                gDeviceType = CL_DEVICE_TYPE_GPU;
            else if( 0 == strcmp( arg, "CL_DEVICE_TYPE_ACCELERATOR" ) )
                gDeviceType = CL_DEVICE_TYPE_ACCELERATOR;
            else if( 0 == strcmp( arg, "CL_DEVICE_TYPE_DEFAULT" ) )
                gDeviceType = CL_DEVICE_TYPE_DEFAULT;
            else
            {
                argList[ argCount ] = arg;
                argCount++;
            }
        }
    }

    if (getenv("CL_WIMPY_MODE")) {
      vlog( "\n" );
      vlog( "*** Detected CL_WIMPY_MODE env                          ***\n" );
      gWimpyMode = 1;
    }

    vlog( "Test binary built %s %s\n", __DATE__, __TIME__ );
    PrintArch();
    PrintDevice();
    if( gWimpyMode )
    {
        vlog( "\n" );
        vlog( "*** WARNING: Testing in Wimpy mode!                     ***\n" );
        vlog( "*** Wimpy mode is not sufficient to verify correctness. ***\n" );
        vlog( "*** It gives warm fuzzy feelings and then nevers calls. ***\n\n" );
        vlog( "*** Wimpy Reduction Factor: %-27u ***\n\n", gWimpyReductionFactor);
    }
    return 0;
}

static void PrintUsage( void )
{
    vlog( "%s [-dthw]: <optional: test names>\n", appName );
    vlog( "\t\t-d\tToggle double precision testing (default: on if double supported)\n" );
    vlog( "\t\t-t\tToggle reporting performance data.\n" );
    vlog( "\t\t-w\tRun in wimpy mode\n" );
    vlog( "\t\t-[2^n]\tSet wimpy reduction factor, recommended range of n is 1-12, default factor(%u)\n", gWimpyReductionFactor);
    vlog( "\t\t-h\tHelp\n" );
    for( int i = 0; i < num_fns; i++ )
    {
        vlog("\t\t%s\n", basefn_names[i] );
    }
}

static void PrintArch( void )
{
    vlog( "sizeof( void*) = %ld\n", sizeof( void *) );

#if defined( __APPLE__ )
#if defined( __ppc__ )
    vlog( "ARCH:\tppc\n" );
#elif defined( __ppc64__ )
    vlog( "ARCH:\tppc64\n" );
#elif defined( __i386__ )
    vlog( "ARCH:\ti386\n" );
#elif defined( __x86_64__ )
    vlog( "ARCH:\tx86_64\n" );
#elif defined( __arm__ )
    vlog( "ARCH:\tarm\n" );
#elif defined( __aarch64__ )
    vlog( "\tARCH:\taarch64\n" );
#else
#error unknown arch
#endif

    int type = 0;
    size_t typeSize = sizeof( type );
    sysctlbyname( "hw.cputype", &type, &typeSize, NULL, 0 );
    vlog( "cpu type:\t%d\n", type );
    typeSize = sizeof( type );
    sysctlbyname( "hw.cpusubtype", &type, &typeSize, NULL, 0 );
    vlog( "cpu subtype:\t%d\n", type );
#endif
}

static void PrintDevice( void)
{
    switch(gDeviceType) {
        case CL_DEVICE_TYPE_CPU:
            vlog( "DEVICE:\tcpu\n" );
            break;
        case CL_DEVICE_TYPE_GPU:
            vlog( "DEVICE:\tgpu\n" );
            break;
        case CL_DEVICE_TYPE_ACCELERATOR:
            vlog( "DEVICE:\taccelerator\n" );
            break;
        default:
            vlog_error( "DEVICE:\tunknown\n" );
            break;
    }
}