Skip to content

Commit

Permalink
reset to default CUDA allocator
Browse files Browse the repository at this point in the history
  • Loading branch information
upsj committed Mar 30, 2023
1 parent ea94c89 commit 33645c0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/utils/executor.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ inline void init_executor(std::shared_ptr<gko::ReferenceExecutor> ref,
throw std::runtime_error{"No suitable CUDA devices"};
}
exec = gko::CudaExecutor::create(
0, ref, std::make_shared<gko::CudaAsyncAllocator>(stream), stream);
0, ref, std::make_shared<gko::CudaAllocator>(stream), stream);
}
}

Expand Down

0 comments on commit 33645c0

Please sign in to comment.