File: 2220-disable-sve-on-arm64.patch

package info (click to toggle)
pytorch-cuda 2.6.0%2Bdfsg-7
  • links: PTS, VCS
  • area: contrib
  • in suites: forky, sid, trixie
  • size: 161,620 kB
  • sloc: python: 1,278,832; cpp: 900,322; ansic: 82,710; asm: 7,754; java: 3,363; sh: 2,811; javascript: 2,443; makefile: 597; ruby: 195; xml: 84; objc: 68
file content (21 lines) | stat: -rw-r--r-- 911 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
Description: Disable use of SVE on arm64
Author: Shengqi Chen <harry@debian.org>
Bug: https://github.com/pytorch/pytorch/issues/146792
Last-Update: 2025-02-24
---
This patch header follows DEP-3: http://dep.debian.net/deps/dep3/
Index: pytorch/cmake/Modules/FindARM.cmake
===================================================================
--- pytorch.orig/cmake/Modules/FindARM.cmake
+++ pytorch/cmake/Modules/FindARM.cmake
@@ -142,7 +142,9 @@ IF(CMAKE_SYSTEM_NAME MATCHES "Linux")
     ENDMACRO()
 
     # Check for SVE256 vector length
-    CHECK_SVE(CXX "SVE256" "-march=armv8-a+sve -msve-vector-bits=256")
+    #CHECK_SVE(CXX "SVE256" "-march=armv8-a+sve -msve-vector-bits=256")
+    SET(CXX_SVE256_FOUND FALSE)
+    message(STATUS "SVE explicitly disabled -- see pytorch issue #146792")
 
     # If SVE256 support is not found, set CXX_SVE_FOUND to FALSE and notify the user
     if(NOT CXX_SVE256_FOUND)