python312Packages.torchWithRocm: remove find_package(rocm-core) workaround

This commit is contained in:
Luna Nova
2025-08-31 19:09:44 -07:00
parent 6c1a925cc2
commit 61e2c0c580
2 changed files with 2 additions and 9 deletions

View File

@@ -369,13 +369,6 @@ buildPythonPackage rec {
substituteInPlace third_party/kineto/libkineto/CMakeLists.txt \
--replace-fail "\''$ENV{ROCM_SOURCE_DIR}" "${rocmtoolkit_joined}"
# Workaround cmake error //include does not exist! in rocm-core-config.cmake
# Removing the call falls back to hip_version. Can likely be removed after ROCm 6.4 bump
substituteInPlace cmake/public/LoadHIP.cmake \
--replace-fail \
"find_package(rocm-core CONFIG)" \
""
# Use composable kernel as dependency, rather than built-in third-party
substituteInPlace aten/src/ATen/CMakeLists.txt \
--replace-fail "list(APPEND ATen_HIP_INCLUDE \''${CMAKE_CURRENT_SOURCE_DIR}/../../../third_party/composable_kernel/include)" "" \

View File

@@ -56,7 +56,7 @@ stdenv.mkDerivation (finalAttrs: {
# Test that the CMake config file can be included and sets expected vars
''
mkdir test_project
cd test_project
pushd test_project
echo '
cmake_minimum_required(VERSION 3.16)
@@ -70,7 +70,7 @@ stdenv.mkDerivation (finalAttrs: {
' > CMakeLists.txt
CMAKE_PREFIX_PATH="$out" cmake .
cd ..
popd
. $out/nix-support/setup-hook
env | grep '^ROCM'