We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7b09af2 commit 588c6acCopy full SHA for 588c6ac
src/libtorch.cc
@@ -1312,12 +1312,12 @@ ModelInstanceState::Execute(
1312
torch::jit::overrideCanFuseOnCPU(false);
1313
torch::jit::overrideCanFuseOnGPU(false);
1314
torch::jit::setTensorExprFuserEnabled(false);
1315
- torch::jit::RegisterCudaFuseGraph::registerPass(true);
+ torch::jit::fuser::cuda::setEnabled(true);
1316
} else {
1317
torch::jit::overrideCanFuseOnCPU(true);
1318
torch::jit::overrideCanFuseOnGPU(true);
1319
torch::jit::setTensorExprFuserEnabled(true);
1320
- torch::jit::RegisterCudaFuseGraph::registerPass(false);
+ torch::jit::fuser::cuda::setEnabled(false);
1321
}
1322
1323
0 commit comments