Commit 80279982 authored by Dan Povey's avatar Dan Povey
Browse files

trunk: minor fix to last trunk commit RE cu-device.cc

git-svn-id: https://svn.code.sf.net/p/kaldi/code/trunk@5243 5e6a8d80-dfce-4ca6-a32a-6e07a63d50c8
parent 65485654
......@@ -126,8 +126,6 @@ void CuDevice::SelectGpuId(std::string use_gpu) {
// Create a CUDA context.
bool got_context = GetCudaContext(num_gpus);
e = cudaThreadSynchronize(); // << CUDA context gets created here.
if (use_gpu != "wait") {
if (!got_context) {
// So far no we don't have context, sleep a bit and retry.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment