startReduction.py

#!/usr/bin/env python3

import argparse, os, sys, subprocess

def which(cmd):
    if sys.platform == 'win32' and '.' not in cmd:
        cmd += '.exe'

    if os.access(cmd, os.F_OK):
        return cmd

    for path in os.environ["PATH"].split(os.pathsep):
        if os.access(os.path.join(path, cmd), os.F_OK):
            return os.path.join(path, cmd)

    return None

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Start C-Reduce for OpenCL kernel.')
    parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
    parser.add_argument('--platform', '-p', help='Platform under test')
    parser.add_argument('--device', '-d', help='Device under test')
    parser.add_argument('--cl-launcher', help='Path to cl_launcher application')
    parser.add_argument('--clang', help='Path to clang application')
    parser.add_argument('--libclc', help='Path to libclc include directory')
    if sys.platform == 'win32':
        parser.add_argument('--oclgrind-platform', help='Platform for Oclgrind')
        parser.add_argument('--oclgrind-device', help='Device for Oclgrind')
    parser.add_argument('test', nargs=1, help='Test script')
    parser.add_argument('kernel', nargs=1, help='OpenCL kernel')

    args = parser.parse_args()

    env = os.environ

    if not args.platform:
        if not env.get('CREDUCE_TEST_PLATFORM'):
            parser.error('No platform specified and CREDUCE_TEST_PLATFORM not defined!')
    else:
        env['CREDUCE_TEST_PLATFORM'] = args.platform

    if not args.device:
        if not env.get('CREDUCE_TEST_DEVICE'):
            parser.error('No device specified and CREDUCE_TEST_DEVICE not defined!')
    else:
        env['CREDUCE_TEST_DEVICE'] = args.device

    if args.cl_launcher:
        env['CREDUCE_TEST_CLLAUNCHER'] = os.path.abspath(args.cl_launcher)

    clLauncher = env.get('CREDUCE_TEST_CLLAUNCHER', os.path.abspath('./cl_launcher'))
    if not which(clLauncher):
        clLauncher = os.path.basename(clLauncher)

        if not which(clLauncher):
            parser.error('No cl-launcher specified, CREDUCE_TEST_CLLAUNCHER not defined and cl_launcher not found!')

    if args.clang:
        env['CREDUCE_TEST_CLANG'] = os.path.abspath(args.clang)

    clang = env.get('CREDUCE_TEST_CLANG', os.path.abspath('./clang'))
    if not which(clang):
        clang = os.path.basename(clang)

        if not which(clang):
            parser.error('No clang specified, CREDUCE_TEST_CLANG not defined and clang not found!')

    if args.libclc:
        env['CREDUCE_LIBCLC_INCLUDE_PATH'] = os.path.abspath(args.libclc)

    if sys.platform == 'win32':
        if not args.oclgrind_platform:
            if not env.get('CREDUCE_TEST_OCLGRIND_PLATFORM'):
                parser.error('No oclgrind-platform specified and CREDUCE_TEST_OCLGRIND_PLATFORM not defined!')
        else:
            env['CREDUCE_TEST_OCLGRIND_PLATFORM'] = args.oclgrind_platform

        if not args.oclgrind_device:
            if not env.get('CREDUCE_TEST_OCLGRIND_DEVICE'):
                parser.error('No oclgrind-device specified and CREDUCE_TEST_OCLGRIND_DEVICE not defined!')
        else:
            env['CREDUCE_TEST_OCLGRIND_DEVICE'] = args.oclgrind_device

    if sys.platform == 'win32':
        creduceArgs = ['perl', '--', which('creduce.pl')]
    else:
        creduceArgs = ['creduce']

    creduceArgs.extend(['-n', '1'])

    if args.verbose:
        creduceArgs.append('--verbose')

    creduceArgs.extend(args.test)
    creduceArgs.extend(args.kernel)

    subprocess.call(creduceArgs, env=env, universal_newlines=True)