@@ -405,10 +405,10 @@ def is_suppressed_dll_file(path_basename: str) -> bool:
405405 return path_basename .startswith (("cudart32_" , "nvvm32" ))
406406
407407
408- # Based on nm output for Linux x86_64 /usr/local/cuda (12.8.1)
408+ # Based on ` nm -D --defined-only` output for Linux x86_64 distributions.
409409EXPECTED_LIB_SYMBOLS = {
410410 "nvJitLink" : (
411- "__nvJitLinkCreate_12_0" , # 12.0 through 12.8 (at least)
411+ "__nvJitLinkCreate_12_0" , # 12.0 through 12.9
412412 "nvJitLinkVersion" , # 12.3 and up
413413 ),
414414 "nvrtc" : ("nvrtcVersion" ,),
@@ -424,16 +424,16 @@ def is_suppressed_dll_file(path_basename: str) -> bool:
424424 "cusolverMg" : ("cusolverMgCreate" ,),
425425 "cusparse" : ("cusparseGetVersion" ,),
426426 "nppc" : ("nppGetLibVersion" ,),
427- "nppial" : ("nppiAdd_32f_C1R " ,),
428- "nppicc" : ("nppiColorToGray_8u_C3C1R " ,),
429- "nppidei" : ("nppiCopy_8u_C1R " ,),
430- "nppif" : ("nppiFilterSobelHorizBorder_8u_C1R " ,),
431- "nppig" : ("nppiResize_8u_C1R " ,),
432- "nppim" : ("nppiErode_8u_C1R " ,),
433- "nppist" : ("nppiMean_8u_C1R " ,),
427+ "nppial" : ("nppiAdd_32f_C1R_Ctx " ,),
428+ "nppicc" : ("nppiColorToGray_8u_C3C1R_Ctx " ,),
429+ "nppidei" : ("nppiCopy_8u_C1R_Ctx " ,),
430+ "nppif" : ("nppiFilterSobelHorizBorder_8u_C1R_Ctx " ,),
431+ "nppig" : ("nppiResize_8u_C1R_Ctx " ,),
432+ "nppim" : ("nppiErode_8u_C1R_Ctx " ,),
433+ "nppist" : ("nppiMean_8u_C1R_Ctx " ,),
434434 "nppisu" : ("nppiFree" ,),
435- "nppitc" : ("nppiThreshold_8u_C1R " ,),
436- "npps" : ("nppsAdd_32f " ,),
435+ "nppitc" : ("nppiThreshold_8u_C1R_Ctx " ,),
436+ "npps" : ("nppsAdd_32f_Ctx " ,),
437437 "nvblas" : ("dgemm" ,),
438438 "cufile" : ("cuFileGetVersion" ,),
439439 # "cufile_rdma": ("rdma_buffer_reg",),
0 commit comments