apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 00:33:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

update flake8 version to support pre-commit in python3.12 (#3000)

Browse Source 
* update flake8 version to support pre-commit in python3.12

* polish code
This commit is contained in:
Zero Rains
2025-07-24 16:43:31 +08:00
committed by GitHub
parent 5151bc92c8
commit 0fb37ab7e4
30 changed files with 324 additions and 275 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
fastdeploy/distributed/custom_all_reduce/cuda_wrapper.py
View File

@@ -113,10 +113,7 @@ class CudaRTLibrary:
Function(
"cudaStreamIsCapturing",
cudaError_t,
[
cudaStream_t,
ctypes.POINTER(cudaStreamCaptureStatus)
]
[cudaStream_t, ctypes.POINTER(cudaStreamCaptureStatus)],
),
]
@@ -197,9 +194,8 @@ class CudaRTLibrary:
self.funcs["cudaIpcOpenMemHandle"](ctypes.byref(devPtr), handle, cudaIpcMemLazyEnablePeerAccess)
)
return devPtr
def cudaStreamIsCapturing(self, stream: cudaStream_t) -> ctypes.c_int:
is_capturing = ctypes.c_int()
self.CUDART_CHECK(
self.funcs["cudaStreamIsCapturing"](stream, is_capturing)
)
self.CUDART_CHECK(self.funcs["cudaStreamIsCapturing"](stream, is_capturing))
return is_capturing