#!/bin/bash

set -e

source /opt/dtk/env.sh
export FORCE_CUDA=1
source /usr/local/bin/fastpt -C

echo "开始编译组件torch-bincount"
python3 setup.py -v bdist_wheel
echo "组件torch-bincount编译完成"

ls dist/
pip install dist/torch_bincount*.whl
pip install dist/torch_bincount*.whl --force-reinstall --no-deps

echo "组件torch-bincount安装完成"


echo "开始torch-bincount单元测试"

python setup.py test