162306a36Sopenharmony_ci/*
262306a36Sopenharmony_ci * Copyright 2021 Advanced Micro Devices, Inc.
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * Permission is hereby granted, free of charge, to any person obtaining a
562306a36Sopenharmony_ci * copy of this software and associated documentation files (the "Software"),
662306a36Sopenharmony_ci * to deal in the Software without restriction, including without limitation
762306a36Sopenharmony_ci * the rights to use, copy, modify, merge, publish, distribute, sublicense,
862306a36Sopenharmony_ci * and/or sell copies of the Software, and to permit persons to whom the
962306a36Sopenharmony_ci * Software is furnished to do so, subject to the following conditions:
1062306a36Sopenharmony_ci *
1162306a36Sopenharmony_ci * The above copyright notice and this permission notice shall be included in
1262306a36Sopenharmony_ci * all copies or substantial portions of the Software.
1362306a36Sopenharmony_ci *
1462306a36Sopenharmony_ci * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1562306a36Sopenharmony_ci * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1662306a36Sopenharmony_ci * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
1762306a36Sopenharmony_ci * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
1862306a36Sopenharmony_ci * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
1962306a36Sopenharmony_ci * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
2062306a36Sopenharmony_ci * OTHER DEALINGS IN THE SOFTWARE.
2162306a36Sopenharmony_ci *
2262306a36Sopenharmony_ci */
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci#include "amdgpu_reset.h"
2562306a36Sopenharmony_ci#include "aldebaran.h"
2662306a36Sopenharmony_ci#include "sienna_cichlid.h"
2762306a36Sopenharmony_ci#include "smu_v13_0_10.h"
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_ciint amdgpu_reset_add_handler(struct amdgpu_reset_control *reset_ctl,
3062306a36Sopenharmony_ci			     struct amdgpu_reset_handler *handler)
3162306a36Sopenharmony_ci{
3262306a36Sopenharmony_ci	/* TODO: Check if handler exists? */
3362306a36Sopenharmony_ci	list_add_tail(&handler->handler_list, &reset_ctl->reset_handlers);
3462306a36Sopenharmony_ci	return 0;
3562306a36Sopenharmony_ci}
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ciint amdgpu_reset_init(struct amdgpu_device *adev)
3862306a36Sopenharmony_ci{
3962306a36Sopenharmony_ci	int ret = 0;
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci	switch (adev->ip_versions[MP1_HWIP][0]) {
4262306a36Sopenharmony_ci	case IP_VERSION(13, 0, 2):
4362306a36Sopenharmony_ci	case IP_VERSION(13, 0, 6):
4462306a36Sopenharmony_ci		ret = aldebaran_reset_init(adev);
4562306a36Sopenharmony_ci		break;
4662306a36Sopenharmony_ci	case IP_VERSION(11, 0, 7):
4762306a36Sopenharmony_ci		ret = sienna_cichlid_reset_init(adev);
4862306a36Sopenharmony_ci		break;
4962306a36Sopenharmony_ci	case IP_VERSION(13, 0, 10):
5062306a36Sopenharmony_ci		ret = smu_v13_0_10_reset_init(adev);
5162306a36Sopenharmony_ci		break;
5262306a36Sopenharmony_ci	default:
5362306a36Sopenharmony_ci		break;
5462306a36Sopenharmony_ci	}
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_ci	return ret;
5762306a36Sopenharmony_ci}
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ciint amdgpu_reset_fini(struct amdgpu_device *adev)
6062306a36Sopenharmony_ci{
6162306a36Sopenharmony_ci	int ret = 0;
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	switch (adev->ip_versions[MP1_HWIP][0]) {
6462306a36Sopenharmony_ci	case IP_VERSION(13, 0, 2):
6562306a36Sopenharmony_ci	case IP_VERSION(13, 0, 6):
6662306a36Sopenharmony_ci		ret = aldebaran_reset_fini(adev);
6762306a36Sopenharmony_ci		break;
6862306a36Sopenharmony_ci	case IP_VERSION(11, 0, 7):
6962306a36Sopenharmony_ci		ret = sienna_cichlid_reset_fini(adev);
7062306a36Sopenharmony_ci		break;
7162306a36Sopenharmony_ci	case IP_VERSION(13, 0, 10):
7262306a36Sopenharmony_ci		ret = smu_v13_0_10_reset_fini(adev);
7362306a36Sopenharmony_ci		break;
7462306a36Sopenharmony_ci	default:
7562306a36Sopenharmony_ci		break;
7662306a36Sopenharmony_ci	}
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	return ret;
7962306a36Sopenharmony_ci}
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ciint amdgpu_reset_prepare_hwcontext(struct amdgpu_device *adev,
8262306a36Sopenharmony_ci				   struct amdgpu_reset_context *reset_context)
8362306a36Sopenharmony_ci{
8462306a36Sopenharmony_ci	struct amdgpu_reset_handler *reset_handler = NULL;
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	if (adev->reset_cntl && adev->reset_cntl->get_reset_handler)
8762306a36Sopenharmony_ci		reset_handler = adev->reset_cntl->get_reset_handler(
8862306a36Sopenharmony_ci			adev->reset_cntl, reset_context);
8962306a36Sopenharmony_ci	if (!reset_handler)
9062306a36Sopenharmony_ci		return -EOPNOTSUPP;
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci	return reset_handler->prepare_hwcontext(adev->reset_cntl,
9362306a36Sopenharmony_ci						reset_context);
9462306a36Sopenharmony_ci}
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ciint amdgpu_reset_perform_reset(struct amdgpu_device *adev,
9762306a36Sopenharmony_ci			       struct amdgpu_reset_context *reset_context)
9862306a36Sopenharmony_ci{
9962306a36Sopenharmony_ci	int ret;
10062306a36Sopenharmony_ci	struct amdgpu_reset_handler *reset_handler = NULL;
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_ci	if (adev->reset_cntl)
10362306a36Sopenharmony_ci		reset_handler = adev->reset_cntl->get_reset_handler(
10462306a36Sopenharmony_ci			adev->reset_cntl, reset_context);
10562306a36Sopenharmony_ci	if (!reset_handler)
10662306a36Sopenharmony_ci		return -EOPNOTSUPP;
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_ci	ret = reset_handler->perform_reset(adev->reset_cntl, reset_context);
10962306a36Sopenharmony_ci	if (ret)
11062306a36Sopenharmony_ci		return ret;
11162306a36Sopenharmony_ci
11262306a36Sopenharmony_ci	return reset_handler->restore_hwcontext(adev->reset_cntl,
11362306a36Sopenharmony_ci						reset_context);
11462306a36Sopenharmony_ci}
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_civoid amdgpu_reset_destroy_reset_domain(struct kref *ref)
11862306a36Sopenharmony_ci{
11962306a36Sopenharmony_ci	struct amdgpu_reset_domain *reset_domain = container_of(ref,
12062306a36Sopenharmony_ci								struct amdgpu_reset_domain,
12162306a36Sopenharmony_ci								refcount);
12262306a36Sopenharmony_ci	if (reset_domain->wq)
12362306a36Sopenharmony_ci		destroy_workqueue(reset_domain->wq);
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	kvfree(reset_domain);
12662306a36Sopenharmony_ci}
12762306a36Sopenharmony_ci
12862306a36Sopenharmony_cistruct amdgpu_reset_domain *amdgpu_reset_create_reset_domain(enum amdgpu_reset_domain_type type,
12962306a36Sopenharmony_ci							     char *wq_name)
13062306a36Sopenharmony_ci{
13162306a36Sopenharmony_ci	struct amdgpu_reset_domain *reset_domain;
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	reset_domain = kvzalloc(sizeof(struct amdgpu_reset_domain), GFP_KERNEL);
13462306a36Sopenharmony_ci	if (!reset_domain) {
13562306a36Sopenharmony_ci		DRM_ERROR("Failed to allocate amdgpu_reset_domain!");
13662306a36Sopenharmony_ci		return NULL;
13762306a36Sopenharmony_ci	}
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	reset_domain->type = type;
14062306a36Sopenharmony_ci	kref_init(&reset_domain->refcount);
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_ci	reset_domain->wq = create_singlethread_workqueue(wq_name);
14362306a36Sopenharmony_ci	if (!reset_domain->wq) {
14462306a36Sopenharmony_ci		DRM_ERROR("Failed to allocate wq for amdgpu_reset_domain!");
14562306a36Sopenharmony_ci		amdgpu_reset_put_reset_domain(reset_domain);
14662306a36Sopenharmony_ci		return NULL;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	}
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ci	atomic_set(&reset_domain->in_gpu_reset, 0);
15162306a36Sopenharmony_ci	atomic_set(&reset_domain->reset_res, 0);
15262306a36Sopenharmony_ci	init_rwsem(&reset_domain->sem);
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	return reset_domain;
15562306a36Sopenharmony_ci}
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_civoid amdgpu_device_lock_reset_domain(struct amdgpu_reset_domain *reset_domain)
15862306a36Sopenharmony_ci{
15962306a36Sopenharmony_ci	atomic_set(&reset_domain->in_gpu_reset, 1);
16062306a36Sopenharmony_ci	down_write(&reset_domain->sem);
16162306a36Sopenharmony_ci}
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci
16462306a36Sopenharmony_civoid amdgpu_device_unlock_reset_domain(struct amdgpu_reset_domain *reset_domain)
16562306a36Sopenharmony_ci{
16662306a36Sopenharmony_ci	atomic_set(&reset_domain->in_gpu_reset, 0);
16762306a36Sopenharmony_ci	up_write(&reset_domain->sem);
16862306a36Sopenharmony_ci}
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci
17162306a36Sopenharmony_ci
172