Skip to content

Commit

Permalink
Windows: Give default AOT targets to Windows build. (#866)
Browse files Browse the repository at this point in the history
Default AOT target list on Windows prefers client GPUs, including ARC,
LNL and MTLs.

Signed-off-by: Feng Yuan <[email protected]>
  • Loading branch information
fengyuan14 committed Sep 5, 2024
1 parent 7b266fc commit d49f1b4
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion cmake/BuildFlags.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,11 @@ if(CMAKE_CXX_COMPILER_ID STREQUAL "GNU" OR CMAKE_CXX_COMPILER_ID STREQUAL "MSVC"
set(SYCL_OFFLINE_COMPILER_CG_OPTIONS "${SYCL_OFFLINE_COMPILER_CG_OPTIONS} -cl-fp32-correctly-rounded-divide-sqrt")
set(SYCL_OFFLINE_COMPILER_CG_OPTIONS "-options '${SYCL_OFFLINE_COMPILER_CG_OPTIONS}'")

set(AOT_TARGETS "pvc,xe-lpg,ats-m150")
if(WIN32)
set(AOT_TARGETS "ats-m150,lnl-m,mtl-u,mtl-h")
else()
set(AOT_TARGETS "pvc,xe-lpg,ats-m150")
endif()
if((DEFINED ENV{TORCH_XPU_ARCH_LIST}) AND NOT ("$ENV{TORCH_XPU_ARCH_LIST}" STREQUAL ""))
set(AOT_TARGETS "$ENV{TORCH_XPU_ARCH_LIST}")
endif()
Expand Down

0 comments on commit d49f1b4

Please sign in to comment.