Searched refs:SessionTask (Results 1 – 3 of 3) sorted by relevance
65 class SessionTask {67 explicit SessionTask(SessionTaskType type, std::shared_ptr<RunOpContext> context) in SessionTask() function69 virtual ~SessionTask() = default;78 class BuildTask : public SessionTask {81 : SessionTask(SessionTaskType::kBuildTask, std::move(context)) {} in BuildTask()87 class LaunchTask : public SessionTask {90 : SessionTask(SessionTaskType::kLaunchTask, std::move(context)) {} in LaunchTask()112 const std::vector<std::shared_ptr<SessionTask>> &GetAllBuildTasks() const { return build_tasks_; } in GetAllBuildTasks()113 …const std::queue<std::shared_ptr<SessionTask>> &GetAllLaunchTasks() const { return launch_tasks_; } in GetAllLaunchTasks()123 std::queue<std::shared_ptr<SessionTask>> empty; in ClearAllResources()[all …]
48 static void BatchBuildKernel(const std::vector<std::shared_ptr<SessionTask>> &build_tasks);
860 void AscendSession::BatchBuildKernel(const std::vector<std::shared_ptr<SessionTask>> &build_tasks) { in BatchBuildKernel()