let mdupdt use compute stream instead of mix stream (#1021)

Former-commit-id: 9a010dae

let mdupdt use compute stream instead of mix stream (#1021)
Former-commit-id: 9a010dae
f7612714 · Jinhui Yuan · GitHub · bab12cc6 · f7612714
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

oneflow/core/graph/normal_model_update_compute_task_node.h oneflow/core/graph/normal_model_update_compute_task_node.h +0 -1

未找到文件。
--- a/oneflow/core/graph/normal_model_update_compute_task_node.h
+++ b/oneflow/core/graph/normal_model_update_compute_task_node.h
@@ -19,7 +19,6 @@ class NormalMdUpdtCompTaskNode final : public CompTaskNode {

  void set_random_seed(uint32_t val) { random_seed_ = val; }
  TaskType GetTaskType() const override { return TaskType::kNormalMdUpdt; }
-  CudaWorkType GetCudaWorkType() const override { return CudaWorkType::kMix; }
  void ToProto(TaskProto*) override;

 private: