Skip to content

Commit 588c6ac

Browse files
authored
Fix deprecation warning (#97)
Signed-off-by: Boris Fomitchev <[email protected]>
1 parent 7b09af2 commit 588c6ac

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/libtorch.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1312,12 +1312,12 @@ ModelInstanceState::Execute(
13121312
torch::jit::overrideCanFuseOnCPU(false);
13131313
torch::jit::overrideCanFuseOnGPU(false);
13141314
torch::jit::setTensorExprFuserEnabled(false);
1315-
torch::jit::RegisterCudaFuseGraph::registerPass(true);
1315+
torch::jit::fuser::cuda::setEnabled(true);
13161316
} else {
13171317
torch::jit::overrideCanFuseOnCPU(true);
13181318
torch::jit::overrideCanFuseOnGPU(true);
13191319
torch::jit::setTensorExprFuserEnabled(true);
1320-
torch::jit::RegisterCudaFuseGraph::registerPass(false);
1320+
torch::jit::fuser::cuda::setEnabled(false);
13211321
}
13221322
}
13231323

0 commit comments

Comments
 (0)