

def rocmnode(name) {
    def node_name = 'rocmtest'
    if(name == 'fiji') {
        node_name = 'rocmtest && fiji';
    } else if(name == 'vega') {
        node_name = 'rocmtest && vega';
    } else if(name == 'vega10') {
        node_name = 'rocmtest && vega10';
    } else if(name == 'vega20') {
        node_name = 'rocmtest && vega20';
    } else if(name == 'gfx908') {
        node_name = 'gfx908';
    } else {
        node_name = name
    }
    return node_name
}



def buildJob(compiler, flags, image, test, testargs){

        env.HSA_ENABLE_SDMA=0 
        checkout scm
        def dockerOpts="--device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined"
        def retimage
        try {
            retimage = docker.build("${image}", "--build-arg PREFIX=/opt/rocm .")
            withDockerContainer(image: image, args: dockerOpts) {
                timeout(time: 5, unit: 'MINUTES')
                {
                    sh 'PATH="/opt/rocm/opencl/bin/x86_64/:$PATH" clinfo'
                }
            }
        } catch(Exception ex) {
            retimage = docker.build("${image}", "--build-arg PREFIX=/opt/rocm --no-cache .")
            withDockerContainer(image: image, args: dockerOpts) {
                timeout(time: 5, unit: 'MINUTES')
                {
                    sh 'PATH="/opt/rocm/opencl/bin/x86_64/:$PATH" clinfo'
                }
            }
        }

        withDockerContainer(image: image, args: dockerOpts + ' -v=/var/jenkins/:/var/jenkins') {
            timeout(time: 6, unit: 'HOURS')
            {
                sh "echo \$HSA_ENABLE_SDMA"
                sh "rm -rf build; mkdir build; cd build; export PATH=/opt/rocm/bin:$PATH; CXX=${compiler} CXXFLAGS='-Werror' cmake -DMIOPEN_GPU_SYNC=On -DMIOPEN_TEST_FLAGS='--disable-verification-cache' -DCMAKE_CXX_FLAGS_DEBUG='-g -fno-omit-frame-pointer -fsanitize=undefined -fno-sanitize-recover=undefined' ${flags} .."
                sh "cd build; CTEST_PARALLEL_LEVEL=4 MIOPEN_VERIFY_CACHE_PATH=/var/jenkins/.cache/miopen/vcache MIOPEN_CONV_PRECISE_ROCBLAS_TIMING=0 dumb-init make -j\$(nproc) ${test}"
                sh "MIOPEN_LOG_LEVEL=5 ./build/bin/${test} ${testargs}"

            }
        }
        return retimage
}


pipeline {
    agent none
    environment{
        image = "miopenrocm"
        buildflag = '-DBUILD_DEV=On -DMIOPEN_TEST_ALL=On -DCMAKE_BUILD_TYPE=releasewithdebug -DCMAKE_PREFIX_PATH=/opt/rocm'
    }
    stages{
        stage("HIP Release Auxilary fp32"){
            parallel{
                stage("Activations") {
                    agent{ label rocmnode("vega")}
                    steps{
                        buildJob("hcc", buildflag, image, 'test_activation', '--all -n 2 --verbose --disable-verification-cache')
                    }
                }

                stage("Tensor Operators") {
                    agent{ label rocmnode("vega") }
                    steps{
                        script { buildJob("hcc", buildflag, image, 'test_tensor_ops', '--all -n 2 --verbose --disable-verification-cache')}
                    }
                }

                stage("Tensor Transpose") {
                    agent{ label rocmnode("vega") }
                    steps{
                        buildJob("hcc", buildflag, image, 'test_tensor_trans', '--all -n 2 --verbose --disable-verification-cache')
                    }
                }

                stage("Tensor Copy") {
                    agent{ label rocmnode("vega") }
                    steps{
                        buildJob("hcc", buildflag, image, 'test_tensor_copy', '--all -n 2 --verbose --disable-verification-cache')
                    }
                }

                stage("Tensor Transform") {
                    agent{ label rocmnode("vega") }
                    steps{
                        buildJob("hcc", buildflag, image, 'test_tensor_transform', '--all -n 2 --verbose --disable-verification-cache')
                    }
                }
            }
        }
    }    
}


