mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-24 02:34:12 +08:00
fix graphs for older cuda (#2328)
This commit is contained in:
parent
ec0d5db67b
commit
8917022deb
@ -54,8 +54,8 @@ void Device::make_current() {
|
||||
|
||||
CommandEncoder::CaptureContext::CaptureContext(CommandEncoder& enc) : enc(enc) {
|
||||
CHECK_CUDA_ERROR(cudaGraphCreate(&graph, 0));
|
||||
CHECK_CUDA_ERROR(cudaStreamBeginCaptureToGraph(
|
||||
enc.stream(), graph, NULL, NULL, 0, cudaStreamCaptureModeGlobal));
|
||||
CHECK_CUDA_ERROR(
|
||||
cudaStreamBeginCapture(enc.stream(), cudaStreamCaptureModeGlobal));
|
||||
}
|
||||
|
||||
CommandEncoder::CaptureContext::~CaptureContext() {
|
||||
|
Loading…
Reference in New Issue
Block a user