Skip to content

Commit

Permalink
use url instead of git for neural_speed.
Browse files Browse the repository at this point in the history
  • Loading branch information
luoyu-intel committed Jan 17, 2024
1 parent 8c583f5 commit 9f0c403
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
1 change: 1 addition & 0 deletions cmake/deps.txt
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ cutlass;https://github.com/NVIDIA/cutlass/archive/refs/tags/v3.1.0.zip;757f90a79
utf8_range;https://github.com/protocolbuffers/utf8_range/archive/72c943dea2b9240cd09efde15191e144bc7c7d38.zip;9925739c9debc0efa2adcb194d371a35b6a03156
extensions;https://github.com/microsoft/onnxruntime-extensions/archive/94142d8391c9791ec71c38336436319a2d4ac7a0.zip;4365ac5140338b4cb75a39944a4be276e3829b3c
composable_kernel;https://github.com/ROCmSoftwarePlatform/composable_kernel/archive/5356c4a943a35e74d7cdc69486afcb8703b9a59a.zip;522382c2af437e09124287e5879ab64af5b2e299
neural_speed;https://github.com/intel/neural-speed/archive/refs/tags/bestlav0.1.1.zip;65b0f7a0d04f72f0d5a8d48af70f0366f2ab3939
10 changes: 5 additions & 5 deletions cmake/external/neural_speed.cmake
Original file line number Diff line number Diff line change
@@ -1,6 +1,3 @@
set(NEURAL_SPEED_URL https://github.com/intel/neural-speed.git)
set(NEURAL_SPEED_TAG 44c05babb9bf01c8b26f8697526f380677cb6800) # kernel-only release v0.1

if ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU" AND onnxruntime_target_platform STREQUAL "x86_64")
set(USE_NEURAL_SPEED TRUE)
elseif ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "MSVC" AND onnxruntime_target_platform STREQUAL "x64")
Expand All @@ -10,9 +7,12 @@ endif()
if(USE_NEURAL_SPEED)
FetchContent_Declare(
neural_speed
GIT_REPOSITORY ${NEURAL_SPEED_URL}
GIT_TAG ${NEURAL_SPEED_TAG}
URL ${DEP_URL_neural_speed}
URL_HASH SHA1=${DEP_SHA1_neural_speed}
)
set(BTLA_USE_OPENMP OFF)
FetchContent_MakeAvailable(neural_speed)
if(NOT neural_speed)
FetchContent_Populate(neural_speed)
endif()
endif()

0 comments on commit 9f0c403

Please sign in to comment.