runnnerrrrrr

1 files changed, 2 insertions, 2 deletions

diff --git a/app/relay/runner.js b/app/relay/runner.js
index b138b87..df1b3f2 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -439,11 +439,11 @@ export function run_next_task(){
 export function stop_task(task, sigkill){
   console.log("stop task", task)
   if (!task) return { error: 'no such task' }
-  if (task === 'cpu' || state.current_cpu_task.task.uuid === task.uuid) {
+  if (task === 'cpu' || (task.uuid && state.current_cpu_task.task.uuid === task.uuid)) {
     console.log('stop cpu task')
     terminate(state.current_cpu_task, sigkill)
     return { status: 'ok' }
-  } else if (task === 'gpu' || state.current_gpu_task.task.uuid === task.uuid) {
+  } else if (task === 'gpu' || (task.uuid && state.current_gpu_task.task.uuid === task.uuid)) {
     console.log('stop gpu task')
     terminate(state.current_gpu_task, sigkill)
     return { status: 'ok' }