genarate and package PDB to wheel.

xuhancn · pytorchmergebot · commit 85ed6addaf03 · 2025-03-03T07:24:22.000Z
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -559,6 +559,9 @@ if(MSVC)
   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /Zc:__cplusplus")
   set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcompiler  /Zc:__cplusplus")
 
+  set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /Zi")
+  set(CMAKE_SHARED_LINKER_FLAGS_RELEASE "${CMAKE_SHARED_LINKER_FLAGS_RELEASE} /DEBUG /OPT:REF /OPT:ICF")
+
   set(CMAKE_NINJA_CMCLDEPS_RC OFF)
   foreach(
     flag_var
diff --git a/setup.py b/setup.py
@@ -1397,6 +1397,7 @@ def main():
                 "lib/*.dylib*",
                 "lib/*.dll",
                 "lib/*.lib",
+                "lib/*.pdb",
             ]
         )
         aotriton_image_path = os.path.join(lib_path, "aotriton.images")

Original file line number	Diff line number	Diff line change
`@@ -1397,6 +1397,7 @@ def main():`
`1397`	`1397`	`"lib/.dylib",`
`1398`	`1398`	`"lib/*.dll",`
`1399`	`1399`	`"lib/*.lib",`
	`1400`	`+ "lib/*.pdb",`
`1400`	`1401`	`]`
`1401`	`1402`	`)`
`1402`	`1403`	`aotriton_image_path = os.path.join(lib_path, "aotriton.images")`