@echo off

call "C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\VC\Auxiliary\Build\vcvars64.bat"

set HSA_OVERRIDE_GFX_VERSION=12.0.0
set PYTORCH_ROCM_ARCH=gfx1200
set ROCM_HOME=C:\Program Files\AMD\ROCm\7.1
set HIP_PATH=%ROCM_HOME%
set PATH=%ROCM_HOME%\bin;%PATH%

set CC=clang-cl
set CXX=clang-cl
set DISTUTILS_USE_SDK=1

:: May help for the command length limit
set SETUPTOOLS_USE_DISTUTILS=local

set MAX_JOBS=8
set CMAKE_BUILD_PARALLEL_LEVEL=8
set FLASH_ATTENTION_FORCE_CK=1

python setup.py build
pause