pipeline {
    agent any
    environment {
        PYPI_PWD = credentials('PYPI_PWD')
        PATH = "/usr/local/clang-7.0.1/bin:/usr/local/cuda/bin/:$PATH"
        LD_LIBRARY_PATH = "/usr/local/clang-7.0.1/lib:/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
        CC = "clang-7"
        CXX = "clang++"
    }
    stages{
        stage('Build') {
            parallel {
                stage('cuda10.0-python3.6') {
                    agent {
                        node {
                            label "cuda10_0 && python3_6"
                            customWorkspace "taichi_cu100_py36"
                        }
                    }
                    environment {
                        PYTHON_EXECUTABLE = "python3.6"
                        CUDA_VERSION = "10.0"
                    }
                    steps{
                        build_taichi()
                    }
                }
                stage('cuda10.0-python3.7') {
                    agent {
                        node {
                            label "cuda10_0 && python3_7"
                            customWorkspace "taichi_cu100_py37"
                        }
                    }
                    environment {
                        PYTHON_EXECUTABLE = "python3.7"
                        CUDA_VERSION = "10.0"
                    }
                    steps{
                        build_taichi()
                    }
                }
                stage('cuda10.0-python3.8') {
                    agent {
                        node {
                            label "cuda10_0 && python3_8"
                            customWorkspace "taichi_cu100_py38"
                        }
                    }
                    environment {
                        PYTHON_EXECUTABLE = "python3.8"
                        CUDA_VERSION = "10.0"
                    }
                    steps{
                        build_taichi()
                    }
                }
            }
        }
        stage('Test') {
            steps {
                sh "echo Testing"
            }
        }
        stage('Release') {
            steps {
                sh "echo releasing"
            }
        }
    }
}

void build_taichi() {
    sh "echo building"
    sh "echo $PATH"
    git 'https://github.com/taichi-dev/taichi.git'
    sh label: '', script: '''
    echo $PATH
    echo $CC
    echo $CXX
    $CC --version
    $CXX --version
    echo $WORKSPACE
    $PYTHON_EXECUTABLE -m pip install twine numpy Pillow scipy pybind11 colorama setuptools astor matplotlib pytest autograd GitPython dill --user
    export TAICHI_REPO_DIR=$WORKSPACE/
    echo $TAICHI_REPO_DIR
    export PYTHONPATH=$TAICHI_REPO_DIR/python
    export PATH=$WORKSPACE/bin/:$PATH
    nvidia-smi
    cd $TAICHI_REPO_DIR
    git submodule update --init --recursive
    [ -e build ] && rm -rf build
    mkdir build && cd build
    export CUDA_BIN_PATH=/usr/local/cuda-${CUDA_VERSION}
    cmake .. -DPYTHON_EXECUTABLE=$PYTHON_EXECUTABLE -DCUDA_VERSION=$CUDA_VERSION
    make -j 8
    ldd libtaichi_core.so
    objdump -T libtaichi_core.so| grep GLIBC
    cd ../python
    ti test -t 1 -na opengl
    $PYTHON_EXECUTABLE build.py upload
    '''
}