Lines Matching refs:cs
232 void cs_get(struct hl_cs *cs) in cs_get() argument
234 kref_get(&cs->refcount); in cs_get()
237 static int cs_get_unless_zero(struct hl_cs *cs) in cs_get_unless_zero() argument
239 return kref_get_unless_zero(&cs->refcount); in cs_get_unless_zero()
242 static void cs_put(struct hl_cs *cs) in cs_put() argument
244 kref_put(&cs->refcount, cs_do_release); in cs_put()
259 bool cs_needs_completion(struct hl_cs *cs) in cs_needs_completion() argument
264 if (cs->staged_cs && !cs->staged_last) in cs_needs_completion()
270 bool cs_needs_timeout(struct hl_cs *cs) in cs_needs_timeout() argument
275 if (cs->staged_cs && !cs->staged_first) in cs_needs_timeout()
310 parser.ctx_id = job->cs->ctx->asid; in cs_parser()
311 parser.cs_sequence = job->cs->sequence; in cs_parser()
322 parser.completion = cs_needs_completion(job->cs); in cs_parser()
351 struct hl_cs *cs = job->cs; in hl_complete_job() local
382 spin_lock(&cs->job_lock); in hl_complete_job()
384 spin_unlock(&cs->job_lock); in hl_complete_job()
400 if (cs_needs_completion(cs) && in hl_complete_job()
407 cs->completion_timestamp = job->timestamp; in hl_complete_job()
409 cs_put(cs); in hl_complete_job()
427 struct hl_cs *cs; in hl_staged_cs_find_first() local
429 list_for_each_entry_reverse(cs, &hdev->cs_mirror_list, mirror_node) in hl_staged_cs_find_first()
430 if (cs->staged_cs && cs->staged_first && in hl_staged_cs_find_first()
431 cs->sequence == cs_seq) in hl_staged_cs_find_first()
432 return cs; in hl_staged_cs_find_first()
444 bool is_staged_cs_last_exists(struct hl_device *hdev, struct hl_cs *cs) in is_staged_cs_last_exists() argument
448 last_entry = list_last_entry(&cs->staged_cs_node, struct hl_cs, in is_staged_cs_last_exists()
467 static void staged_cs_get(struct hl_device *hdev, struct hl_cs *cs) in staged_cs_get() argument
474 if (!cs->staged_last) in staged_cs_get()
475 cs_get(cs); in staged_cs_get()
486 static void staged_cs_put(struct hl_device *hdev, struct hl_cs *cs) in staged_cs_put() argument
491 if (!cs_needs_completion(cs)) in staged_cs_put()
492 cs_put(cs); in staged_cs_put()
495 static void cs_handle_tdr(struct hl_device *hdev, struct hl_cs *cs) in cs_handle_tdr() argument
499 if (!cs_needs_timeout(cs)) in cs_handle_tdr()
512 if (cs->staged_cs && cs->staged_last) { in cs_handle_tdr()
513 first_cs = hl_staged_cs_find_first(hdev, cs->staged_sequence); in cs_handle_tdr()
515 cs = first_cs; in cs_handle_tdr()
523 if (cs->timedout || hdev->timeout_jiffies == MAX_SCHEDULE_TIMEOUT) in cs_handle_tdr()
526 if (cs->tdr_active) in cs_handle_tdr()
527 cancel_delayed_work_sync(&cs->work_tdr); in cs_handle_tdr()
595 static void complete_multi_cs(struct hl_device *hdev, struct hl_cs *cs) in complete_multi_cs() argument
597 struct hl_fence *fence = cs->fence; in complete_multi_cs()
601 if (cs->staged_cs && !cs->staged_first) in complete_multi_cs()
645 struct hl_cs *cs, in cs_release_sob_reset_handler() argument
652 if (!hl_cs_cmpl->hw_sob || !cs->submitted) in cs_release_sob_reset_handler()
685 struct hl_cs *cs = container_of(ref, struct hl_cs, refcount); in cs_do_release() local
686 struct hl_device *hdev = cs->ctx->hdev; in cs_do_release()
689 container_of(cs->fence, struct hl_cs_compl, base_fence); in cs_do_release()
691 cs->completed = true; in cs_do_release()
701 list_for_each_entry_safe(job, tmp, &cs->job_list, cs_node) in cs_do_release()
704 if (!cs->submitted) { in cs_do_release()
710 if (cs->type == CS_TYPE_WAIT || in cs_do_release()
711 cs->type == CS_TYPE_COLLECTIVE_WAIT) in cs_do_release()
712 hl_fence_put(cs->signal_fence); in cs_do_release()
718 hl_hw_queue_update_ci(cs); in cs_do_release()
722 list_del_init(&cs->mirror_node); in cs_do_release()
725 cs_handle_tdr(hdev, cs); in cs_do_release()
727 if (cs->staged_cs) { in cs_do_release()
731 if (cs->staged_last) { in cs_do_release()
735 &cs->staged_cs_node, staged_cs_node) in cs_do_release()
743 if (cs->submitted) { in cs_do_release()
745 list_del(&cs->staged_cs_node); in cs_do_release()
757 if ((cs->type == CS_TYPE_WAIT || cs->type == CS_TYPE_COLLECTIVE_WAIT) && cs->encaps_signals) in cs_do_release()
758 kref_put(&cs->encaps_sig_hdl->refcount, hl_encaps_release_handle_and_put_ctx); in cs_do_release()
764 hl_debugfs_remove_cs(cs); in cs_do_release()
766 hdev->shadow_cs_queue[cs->sequence & (hdev->asic_prop.max_pending_cs - 1)] = NULL; in cs_do_release()
772 if (cs->timedout) in cs_do_release()
773 cs->fence->error = -ETIMEDOUT; in cs_do_release()
774 else if (cs->aborted) in cs_do_release()
775 cs->fence->error = -EIO; in cs_do_release()
776 else if (!cs->submitted) in cs_do_release()
777 cs->fence->error = -EBUSY; in cs_do_release()
779 if (unlikely(cs->skip_reset_on_timeout)) { in cs_do_release()
782 cs->sequence, in cs_do_release()
783 div_u64(jiffies - cs->submission_time_jiffies, HZ)); in cs_do_release()
786 if (cs->timestamp) { in cs_do_release()
787 cs->fence->timestamp = cs->completion_timestamp; in cs_do_release()
788 hl_push_cs_outcome(hdev, &cs->ctx->outcome_store, cs->sequence, in cs_do_release()
789 cs->fence->timestamp, cs->fence->error); in cs_do_release()
792 hl_ctx_put(cs->ctx); in cs_do_release()
794 complete_all(&cs->fence->completion); in cs_do_release()
795 complete_multi_cs(hdev, cs); in cs_do_release()
797 cs_release_sob_reset_handler(hdev, cs, hl_cs_cmpl); in cs_do_release()
799 hl_fence_put(cs->fence); in cs_do_release()
801 kfree(cs->jobs_in_queue_cnt); in cs_do_release()
802 kfree(cs); in cs_do_release()
810 struct hl_cs *cs = container_of(work, struct hl_cs, in cs_timedout() local
812 bool skip_reset_on_timeout = cs->skip_reset_on_timeout, device_reset = false; in cs_timedout()
814 rc = cs_get_unless_zero(cs); in cs_timedout()
818 if ((!cs->submitted) || (cs->completed)) { in cs_timedout()
819 cs_put(cs); in cs_timedout()
823 hdev = cs->ctx->hdev; in cs_timedout()
832 cs->timedout = true; in cs_timedout()
839 hdev->captured_err_info.cs_timeout.seq = cs->sequence; in cs_timedout()
843 switch (cs->type) { in cs_timedout()
847 cs->sequence); in cs_timedout()
853 cs->sequence); in cs_timedout()
859 cs->sequence); in cs_timedout()
865 cs->sequence); in cs_timedout()
873 cs_put(cs); in cs_timedout()
890 struct hl_cs *cs; in allocate_cs() local
895 cs = kzalloc(sizeof(*cs), GFP_ATOMIC); in allocate_cs()
896 if (!cs) in allocate_cs()
897 cs = kzalloc(sizeof(*cs), GFP_KERNEL); in allocate_cs()
899 if (!cs) { in allocate_cs()
908 cs->ctx = ctx; in allocate_cs()
909 cs->submitted = false; in allocate_cs()
910 cs->completed = false; in allocate_cs()
911 cs->type = cs_type; in allocate_cs()
912 cs->timestamp = !!(flags & HL_CS_FLAGS_TIMESTAMP); in allocate_cs()
913 cs->encaps_signals = !!(flags & HL_CS_FLAGS_ENCAP_SIGNALS); in allocate_cs()
914 cs->timeout_jiffies = timeout; in allocate_cs()
915 cs->skip_reset_on_timeout = in allocate_cs()
918 cs->submission_time_jiffies = jiffies; in allocate_cs()
919 INIT_LIST_HEAD(&cs->job_list); in allocate_cs()
920 INIT_DELAYED_WORK(&cs->work_tdr, cs_timedout); in allocate_cs()
921 kref_init(&cs->refcount); in allocate_cs()
922 spin_lock_init(&cs->job_lock); in allocate_cs()
935 cs->jobs_in_queue_cnt = kcalloc(hdev->asic_prop.max_queues, in allocate_cs()
936 sizeof(*cs->jobs_in_queue_cnt), GFP_ATOMIC); in allocate_cs()
937 if (!cs->jobs_in_queue_cnt) in allocate_cs()
938 cs->jobs_in_queue_cnt = kcalloc(hdev->asic_prop.max_queues, in allocate_cs()
939 sizeof(*cs->jobs_in_queue_cnt), GFP_KERNEL); in allocate_cs()
941 if (!cs->jobs_in_queue_cnt) { in allocate_cs()
949 cs_cmpl->type = cs->type; in allocate_cs()
951 cs->fence = &cs_cmpl->base_fence; in allocate_cs()
983 cs->sequence = cs_cmpl->cs_seq; in allocate_cs()
996 *cs_new = cs; in allocate_cs()
1002 kfree(cs->jobs_in_queue_cnt); in allocate_cs()
1006 kfree(cs); in allocate_cs()
1011 static void cs_rollback(struct hl_device *hdev, struct hl_cs *cs) in cs_rollback() argument
1015 staged_cs_put(hdev, cs); in cs_rollback()
1017 list_for_each_entry_safe(job, tmp, &cs->job_list, cs_node) in cs_rollback()
1052 struct hl_cs *cs, *tmp; in hl_cs_rollback_all() local
1067 list_for_each_entry_safe(cs, tmp, &hdev->cs_mirror_list, mirror_node) { in hl_cs_rollback_all()
1068 cs_get(cs); in hl_cs_rollback_all()
1069 cs->aborted = true; in hl_cs_rollback_all()
1071 cs->ctx->asid, cs->sequence); in hl_cs_rollback_all()
1072 cs_rollback(hdev, cs); in hl_cs_rollback_all()
1073 cs_put(cs); in hl_cs_rollback_all()
1131 struct hl_cs *cs; in force_complete_cs() local
1135 list_for_each_entry(cs, &hdev->cs_mirror_list, mirror_node) { in force_complete_cs()
1136 cs->fence->error = -EIO; in force_complete_cs()
1137 complete_all(&cs->fence->completion); in force_complete_cs()
1154 struct hl_cs *cs = job->cs; in job_wq_completion() local
1155 struct hl_device *hdev = cs->ctx->hdev; in job_wq_completion()
1163 struct hl_cs *cs = container_of(work, struct hl_cs, finish_work); in cs_completion() local
1164 struct hl_device *hdev = cs->ctx->hdev; in cs_completion()
1167 list_for_each_entry_safe(job, tmp, &cs->job_list, cs_node) in cs_completion()
1413 static int cs_staged_submission(struct hl_device *hdev, struct hl_cs *cs, in cs_staged_submission() argument
1420 cs->staged_last = !!(flags & HL_CS_FLAGS_STAGED_SUBMISSION_LAST); in cs_staged_submission()
1421 cs->staged_first = !!(flags & HL_CS_FLAGS_STAGED_SUBMISSION_FIRST); in cs_staged_submission()
1423 if (cs->staged_first) { in cs_staged_submission()
1425 INIT_LIST_HEAD(&cs->staged_cs_node); in cs_staged_submission()
1426 cs->staged_sequence = cs->sequence; in cs_staged_submission()
1428 if (cs->encaps_signals) in cs_staged_submission()
1429 cs->encaps_sig_hdl_id = encaps_signal_handle; in cs_staged_submission()
1434 cs->staged_sequence = sequence; in cs_staged_submission()
1438 staged_cs_get(hdev, cs); in cs_staged_submission()
1440 cs->staged_cs = true; in cs_staged_submission()
1467 struct hl_cs *cs; in cs_ioctl_default() local
1489 staged_mid ? user_sequence : ULLONG_MAX, &cs, flags, in cs_ioctl_default()
1494 *cs_seq = cs->sequence; in cs_ioctl_default()
1496 hl_debugfs_add_cs(cs); in cs_ioctl_default()
1498 rc = cs_staged_submission(hdev, cs, user_sequence, flags, in cs_ioctl_default()
1506 if (cs->staged_cs) in cs_ioctl_default()
1507 *cs_seq = cs->staged_sequence; in cs_ioctl_default()
1567 job->cs = cs; in cs_ioctl_default()
1572 cs->jobs_in_queue_cnt[job->hw_queue_id]++; in cs_ioctl_default()
1573 cs->jobs_cnt++; in cs_ioctl_default()
1575 list_add_tail(&job->cs_node, &cs->job_list); in cs_ioctl_default()
1583 if (cs_needs_completion(cs) && in cs_ioctl_default()
1586 cs_get(cs); in cs_ioctl_default()
1596 cs->ctx->asid, cs->sequence, job->id, rc); in cs_ioctl_default()
1604 if (int_queues_only && cs_needs_completion(cs)) { in cs_ioctl_default()
1609 cs->ctx->asid, cs->sequence); in cs_ioctl_default()
1615 INIT_WORK(&cs->finish_work, cs_completion); in cs_ioctl_default()
1622 cs->fence->stream_master_qid_map = stream_master_qid_map; in cs_ioctl_default()
1624 rc = hl_hw_queue_schedule_cs(cs); in cs_ioctl_default()
1629 cs->ctx->asid, cs->sequence, rc); in cs_ioctl_default()
1633 *signal_initial_sob_count = cs->initial_sob_count; in cs_ioctl_default()
1642 cs_rollback(hdev, cs); in cs_ioctl_default()
1647 cs_put(cs); in cs_ioctl_default()
1911 struct hl_ctx *ctx, struct hl_cs *cs, in cs_ioctl_signal_wait_create_jobs() argument
1929 if (cs->type == CS_TYPE_WAIT) in cs_ioctl_signal_wait_create_jobs()
1944 job->cs = cs; in cs_ioctl_signal_wait_create_jobs()
1950 if ((cs->type == CS_TYPE_WAIT || cs->type == CS_TYPE_COLLECTIVE_WAIT) in cs_ioctl_signal_wait_create_jobs()
1951 && cs->encaps_signals) in cs_ioctl_signal_wait_create_jobs()
1964 cs_get(cs); in cs_ioctl_signal_wait_create_jobs()
1966 cs->jobs_in_queue_cnt[job->hw_queue_id]++; in cs_ioctl_signal_wait_create_jobs()
1967 cs->jobs_cnt++; in cs_ioctl_signal_wait_create_jobs()
1969 list_add_tail(&job->cs_node, &cs->job_list); in cs_ioctl_signal_wait_create_jobs()
2184 struct hl_cs *cs; in cs_ioctl_signal_wait() local
2344 rc = allocate_cs(hdev, ctx, cs_type, ULLONG_MAX, &cs, flags, timeout); in cs_ioctl_signal_wait()
2359 cs->signal_fence = sig_fence; in cs_ioctl_signal_wait()
2364 if (cs->encaps_signals) in cs_ioctl_signal_wait()
2365 cs->encaps_sig_hdl = encaps_sig_hdl; in cs_ioctl_signal_wait()
2368 hl_debugfs_add_cs(cs); in cs_ioctl_signal_wait()
2370 *cs_seq = cs->sequence; in cs_ioctl_signal_wait()
2373 rc = cs_ioctl_signal_wait_create_jobs(hdev, ctx, cs, q_type, in cs_ioctl_signal_wait()
2377 cs, q_idx, collective_engine_id, in cs_ioctl_signal_wait()
2389 INIT_WORK(&cs->finish_work, cs_completion); in cs_ioctl_signal_wait()
2391 rc = hl_hw_queue_schedule_cs(cs); in cs_ioctl_signal_wait()
2402 ctx->asid, cs->sequence, rc); in cs_ioctl_signal_wait()
2406 *signal_sob_addr_offset = cs->sob_addr_offset; in cs_ioctl_signal_wait()
2407 *signal_initial_sob_count = cs->initial_sob_count; in cs_ioctl_signal_wait()
2415 cs_rollback(hdev, cs); in cs_ioctl_signal_wait()
2420 cs_put(cs); in cs_ioctl_signal_wait()