Fix device ids not being properly assigned to python kernels

scanner-research · Apr 16, 2019 · f56adf3 · f56adf3
1 parent ab71156
commit f56adf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scanner/engine/python_kernel.cpp b/scanner/engine/python_kernel.cpp
@@ -61,7 +61,7 @@ PythonKernel::PythonKernel(const KernelConfig &config,
   send_pipe_name_ = tfm::format("%s_send_pipe_%s", op_name_, rand);
   recv_pipe_name_ = tfm::format("%s_recv_pipe_%s", op_name_, rand);
   kernel_name_ = tfm::format("%s_kernel_%s", op_name_, rand);
-  std::string kernel_ns_name_ = tfm::format("ns_%s", op_name_);
+  std::string kernel_ns_name_ = tfm::format("ns_%s_%s", op_name_, rand);
 
   try {
     py::module main = py::module::import("__main__");