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 "sienna_cichlid.h"
2562306a36Sopenharmony_ci#include "amdgpu_reset.h"
2662306a36Sopenharmony_ci#include "amdgpu_amdkfd.h"
2762306a36Sopenharmony_ci#include "amdgpu_dpm.h"
2862306a36Sopenharmony_ci#include "amdgpu_job.h"
2962306a36Sopenharmony_ci#include "amdgpu_ring.h"
3062306a36Sopenharmony_ci#include "amdgpu_ras.h"
3162306a36Sopenharmony_ci#include "amdgpu_psp.h"
3262306a36Sopenharmony_ci#include "amdgpu_xgmi.h"
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_cistatic bool sienna_cichlid_is_mode2_default(struct amdgpu_reset_control *reset_ctl)
3562306a36Sopenharmony_ci{
3662306a36Sopenharmony_ci#if 0
3762306a36Sopenharmony_ci	struct amdgpu_device *adev = (struct amdgpu_device *)reset_ctl->handle;
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	if (adev->ip_versions[MP1_HWIP][0] == IP_VERSION(11, 0, 7) &&
4062306a36Sopenharmony_ci	    adev->pm.fw_version >= 0x3a5500 && !amdgpu_sriov_vf(adev))
4162306a36Sopenharmony_ci		return true;
4262306a36Sopenharmony_ci#endif
4362306a36Sopenharmony_ci	return amdgpu_reset_method == AMD_RESET_METHOD_MODE2;
4462306a36Sopenharmony_ci}
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_cistatic struct amdgpu_reset_handler *
4762306a36Sopenharmony_cisienna_cichlid_get_reset_handler(struct amdgpu_reset_control *reset_ctl,
4862306a36Sopenharmony_ci			    struct amdgpu_reset_context *reset_context)
4962306a36Sopenharmony_ci{
5062306a36Sopenharmony_ci	struct amdgpu_reset_handler *handler;
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	if (reset_context->method != AMD_RESET_METHOD_NONE) {
5362306a36Sopenharmony_ci		list_for_each_entry(handler, &reset_ctl->reset_handlers,
5462306a36Sopenharmony_ci				     handler_list) {
5562306a36Sopenharmony_ci			if (handler->reset_method == reset_context->method)
5662306a36Sopenharmony_ci				return handler;
5762306a36Sopenharmony_ci		}
5862306a36Sopenharmony_ci	}
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_ci	if (sienna_cichlid_is_mode2_default(reset_ctl)) {
6162306a36Sopenharmony_ci		list_for_each_entry (handler, &reset_ctl->reset_handlers,
6262306a36Sopenharmony_ci				     handler_list) {
6362306a36Sopenharmony_ci			if (handler->reset_method == AMD_RESET_METHOD_MODE2)
6462306a36Sopenharmony_ci				return handler;
6562306a36Sopenharmony_ci		}
6662306a36Sopenharmony_ci	}
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_ci	return NULL;
6962306a36Sopenharmony_ci}
7062306a36Sopenharmony_ci
7162306a36Sopenharmony_cistatic int sienna_cichlid_mode2_suspend_ip(struct amdgpu_device *adev)
7262306a36Sopenharmony_ci{
7362306a36Sopenharmony_ci	int r, i;
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_ci	amdgpu_device_set_pg_state(adev, AMD_PG_STATE_UNGATE);
7662306a36Sopenharmony_ci	amdgpu_device_set_cg_state(adev, AMD_CG_STATE_UNGATE);
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	for (i = adev->num_ip_blocks - 1; i >= 0; i--) {
7962306a36Sopenharmony_ci		if (!(adev->ip_blocks[i].version->type ==
8062306a36Sopenharmony_ci			      AMD_IP_BLOCK_TYPE_GFX ||
8162306a36Sopenharmony_ci		      adev->ip_blocks[i].version->type ==
8262306a36Sopenharmony_ci			      AMD_IP_BLOCK_TYPE_SDMA))
8362306a36Sopenharmony_ci			continue;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci		r = adev->ip_blocks[i].version->funcs->suspend(adev);
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci		if (r) {
8862306a36Sopenharmony_ci			dev_err(adev->dev,
8962306a36Sopenharmony_ci				"suspend of IP block <%s> failed %d\n",
9062306a36Sopenharmony_ci				adev->ip_blocks[i].version->funcs->name, r);
9162306a36Sopenharmony_ci			return r;
9262306a36Sopenharmony_ci		}
9362306a36Sopenharmony_ci		adev->ip_blocks[i].status.hw = false;
9462306a36Sopenharmony_ci	}
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	return r;
9762306a36Sopenharmony_ci}
9862306a36Sopenharmony_ci
9962306a36Sopenharmony_cistatic int
10062306a36Sopenharmony_cisienna_cichlid_mode2_prepare_hwcontext(struct amdgpu_reset_control *reset_ctl,
10162306a36Sopenharmony_ci				  struct amdgpu_reset_context *reset_context)
10262306a36Sopenharmony_ci{
10362306a36Sopenharmony_ci	int r = 0;
10462306a36Sopenharmony_ci	struct amdgpu_device *adev = (struct amdgpu_device *)reset_ctl->handle;
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	if (!amdgpu_sriov_vf(adev)) {
10762306a36Sopenharmony_ci		if (adev->gfxhub.funcs->mode2_save_regs)
10862306a36Sopenharmony_ci			adev->gfxhub.funcs->mode2_save_regs(adev);
10962306a36Sopenharmony_ci		if (adev->gfxhub.funcs->halt)
11062306a36Sopenharmony_ci			adev->gfxhub.funcs->halt(adev);
11162306a36Sopenharmony_ci		r = sienna_cichlid_mode2_suspend_ip(adev);
11262306a36Sopenharmony_ci	}
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci	return r;
11562306a36Sopenharmony_ci}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_cistatic void sienna_cichlid_async_reset(struct work_struct *work)
11862306a36Sopenharmony_ci{
11962306a36Sopenharmony_ci	struct amdgpu_reset_handler *handler;
12062306a36Sopenharmony_ci	struct amdgpu_reset_control *reset_ctl =
12162306a36Sopenharmony_ci		container_of(work, struct amdgpu_reset_control, reset_work);
12262306a36Sopenharmony_ci	struct amdgpu_device *adev = (struct amdgpu_device *)reset_ctl->handle;
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci	list_for_each_entry(handler, &reset_ctl->reset_handlers,
12562306a36Sopenharmony_ci			     handler_list) {
12662306a36Sopenharmony_ci		if (handler->reset_method == reset_ctl->active_reset) {
12762306a36Sopenharmony_ci			dev_dbg(adev->dev, "Resetting device\n");
12862306a36Sopenharmony_ci			handler->do_reset(adev);
12962306a36Sopenharmony_ci			break;
13062306a36Sopenharmony_ci		}
13162306a36Sopenharmony_ci	}
13262306a36Sopenharmony_ci}
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_cistatic int sienna_cichlid_mode2_reset(struct amdgpu_device *adev)
13562306a36Sopenharmony_ci{
13662306a36Sopenharmony_ci	/* disable BM */
13762306a36Sopenharmony_ci	pci_clear_master(adev->pdev);
13862306a36Sopenharmony_ci	return amdgpu_dpm_mode2_reset(adev);
13962306a36Sopenharmony_ci}
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_cistatic int
14262306a36Sopenharmony_cisienna_cichlid_mode2_perform_reset(struct amdgpu_reset_control *reset_ctl,
14362306a36Sopenharmony_ci			      struct amdgpu_reset_context *reset_context)
14462306a36Sopenharmony_ci{
14562306a36Sopenharmony_ci	struct amdgpu_device *adev = (struct amdgpu_device *)reset_ctl->handle;
14662306a36Sopenharmony_ci	int r;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	r = sienna_cichlid_mode2_reset(adev);
14962306a36Sopenharmony_ci	if (r) {
15062306a36Sopenharmony_ci		dev_err(adev->dev,
15162306a36Sopenharmony_ci			"ASIC reset failed with error, %d ", r);
15262306a36Sopenharmony_ci	}
15362306a36Sopenharmony_ci	return r;
15462306a36Sopenharmony_ci}
15562306a36Sopenharmony_ci
15662306a36Sopenharmony_cistatic int sienna_cichlid_mode2_restore_ip(struct amdgpu_device *adev)
15762306a36Sopenharmony_ci{
15862306a36Sopenharmony_ci	int i, r;
15962306a36Sopenharmony_ci	struct psp_context *psp = &adev->psp;
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci	r = psp_rlc_autoload_start(psp);
16262306a36Sopenharmony_ci	if (r) {
16362306a36Sopenharmony_ci		dev_err(adev->dev, "Failed to start rlc autoload\n");
16462306a36Sopenharmony_ci		return r;
16562306a36Sopenharmony_ci	}
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_ci	/* Reinit GFXHUB */
16862306a36Sopenharmony_ci	if (adev->gfxhub.funcs->mode2_restore_regs)
16962306a36Sopenharmony_ci		adev->gfxhub.funcs->mode2_restore_regs(adev);
17062306a36Sopenharmony_ci	adev->gfxhub.funcs->init(adev);
17162306a36Sopenharmony_ci	r = adev->gfxhub.funcs->gart_enable(adev);
17262306a36Sopenharmony_ci	if (r) {
17362306a36Sopenharmony_ci		dev_err(adev->dev, "GFXHUB gart reenable failed after reset\n");
17462306a36Sopenharmony_ci		return r;
17562306a36Sopenharmony_ci	}
17662306a36Sopenharmony_ci
17762306a36Sopenharmony_ci	for (i = 0; i < adev->num_ip_blocks; i++) {
17862306a36Sopenharmony_ci		if (adev->ip_blocks[i].version->type == AMD_IP_BLOCK_TYPE_IH) {
17962306a36Sopenharmony_ci			r = adev->ip_blocks[i].version->funcs->resume(adev);
18062306a36Sopenharmony_ci			if (r) {
18162306a36Sopenharmony_ci				dev_err(adev->dev,
18262306a36Sopenharmony_ci					"resume of IP block <%s> failed %d\n",
18362306a36Sopenharmony_ci					adev->ip_blocks[i].version->funcs->name, r);
18462306a36Sopenharmony_ci				return r;
18562306a36Sopenharmony_ci			}
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_ci			adev->ip_blocks[i].status.hw = true;
18862306a36Sopenharmony_ci		}
18962306a36Sopenharmony_ci	}
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	for (i = 0; i < adev->num_ip_blocks; i++) {
19262306a36Sopenharmony_ci		if (!(adev->ip_blocks[i].version->type ==
19362306a36Sopenharmony_ci			      AMD_IP_BLOCK_TYPE_GFX ||
19462306a36Sopenharmony_ci		      adev->ip_blocks[i].version->type ==
19562306a36Sopenharmony_ci			      AMD_IP_BLOCK_TYPE_SDMA))
19662306a36Sopenharmony_ci			continue;
19762306a36Sopenharmony_ci		r = adev->ip_blocks[i].version->funcs->resume(adev);
19862306a36Sopenharmony_ci		if (r) {
19962306a36Sopenharmony_ci			dev_err(adev->dev,
20062306a36Sopenharmony_ci				"resume of IP block <%s> failed %d\n",
20162306a36Sopenharmony_ci				adev->ip_blocks[i].version->funcs->name, r);
20262306a36Sopenharmony_ci			return r;
20362306a36Sopenharmony_ci		}
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_ci		adev->ip_blocks[i].status.hw = true;
20662306a36Sopenharmony_ci	}
20762306a36Sopenharmony_ci
20862306a36Sopenharmony_ci	for (i = 0; i < adev->num_ip_blocks; i++) {
20962306a36Sopenharmony_ci		if (!(adev->ip_blocks[i].version->type ==
21062306a36Sopenharmony_ci			      AMD_IP_BLOCK_TYPE_GFX ||
21162306a36Sopenharmony_ci		      adev->ip_blocks[i].version->type ==
21262306a36Sopenharmony_ci			      AMD_IP_BLOCK_TYPE_SDMA))
21362306a36Sopenharmony_ci			continue;
21462306a36Sopenharmony_ci
21562306a36Sopenharmony_ci		if (adev->ip_blocks[i].version->funcs->late_init) {
21662306a36Sopenharmony_ci			r = adev->ip_blocks[i].version->funcs->late_init(
21762306a36Sopenharmony_ci				(void *)adev);
21862306a36Sopenharmony_ci			if (r) {
21962306a36Sopenharmony_ci				dev_err(adev->dev,
22062306a36Sopenharmony_ci					"late_init of IP block <%s> failed %d after reset\n",
22162306a36Sopenharmony_ci					adev->ip_blocks[i].version->funcs->name,
22262306a36Sopenharmony_ci					r);
22362306a36Sopenharmony_ci				return r;
22462306a36Sopenharmony_ci			}
22562306a36Sopenharmony_ci		}
22662306a36Sopenharmony_ci		adev->ip_blocks[i].status.late_initialized = true;
22762306a36Sopenharmony_ci	}
22862306a36Sopenharmony_ci
22962306a36Sopenharmony_ci	amdgpu_device_set_cg_state(adev, AMD_CG_STATE_GATE);
23062306a36Sopenharmony_ci	amdgpu_device_set_pg_state(adev, AMD_PG_STATE_GATE);
23162306a36Sopenharmony_ci
23262306a36Sopenharmony_ci	return r;
23362306a36Sopenharmony_ci}
23462306a36Sopenharmony_ci
23562306a36Sopenharmony_cistatic int
23662306a36Sopenharmony_cisienna_cichlid_mode2_restore_hwcontext(struct amdgpu_reset_control *reset_ctl,
23762306a36Sopenharmony_ci				  struct amdgpu_reset_context *reset_context)
23862306a36Sopenharmony_ci{
23962306a36Sopenharmony_ci	int r;
24062306a36Sopenharmony_ci	struct amdgpu_device *tmp_adev = (struct amdgpu_device *)reset_ctl->handle;
24162306a36Sopenharmony_ci
24262306a36Sopenharmony_ci	dev_info(tmp_adev->dev,
24362306a36Sopenharmony_ci			"GPU reset succeeded, trying to resume\n");
24462306a36Sopenharmony_ci	r = sienna_cichlid_mode2_restore_ip(tmp_adev);
24562306a36Sopenharmony_ci	if (r)
24662306a36Sopenharmony_ci		goto end;
24762306a36Sopenharmony_ci
24862306a36Sopenharmony_ci	/*
24962306a36Sopenharmony_ci	* Add this ASIC as tracked as reset was already
25062306a36Sopenharmony_ci	* complete successfully.
25162306a36Sopenharmony_ci	*/
25262306a36Sopenharmony_ci	amdgpu_register_gpu_instance(tmp_adev);
25362306a36Sopenharmony_ci
25462306a36Sopenharmony_ci	/* Resume RAS */
25562306a36Sopenharmony_ci	amdgpu_ras_resume(tmp_adev);
25662306a36Sopenharmony_ci
25762306a36Sopenharmony_ci	amdgpu_irq_gpu_reset_resume_helper(tmp_adev);
25862306a36Sopenharmony_ci
25962306a36Sopenharmony_ci	r = amdgpu_ib_ring_tests(tmp_adev);
26062306a36Sopenharmony_ci	if (r) {
26162306a36Sopenharmony_ci		dev_err(tmp_adev->dev,
26262306a36Sopenharmony_ci			"ib ring test failed (%d).\n", r);
26362306a36Sopenharmony_ci		r = -EAGAIN;
26462306a36Sopenharmony_ci		goto end;
26562306a36Sopenharmony_ci	}
26662306a36Sopenharmony_ci
26762306a36Sopenharmony_ciend:
26862306a36Sopenharmony_ci	if (r)
26962306a36Sopenharmony_ci		return -EAGAIN;
27062306a36Sopenharmony_ci	else
27162306a36Sopenharmony_ci		return r;
27262306a36Sopenharmony_ci}
27362306a36Sopenharmony_ci
27462306a36Sopenharmony_cistatic struct amdgpu_reset_handler sienna_cichlid_mode2_handler = {
27562306a36Sopenharmony_ci	.reset_method		= AMD_RESET_METHOD_MODE2,
27662306a36Sopenharmony_ci	.prepare_env		= NULL,
27762306a36Sopenharmony_ci	.prepare_hwcontext	= sienna_cichlid_mode2_prepare_hwcontext,
27862306a36Sopenharmony_ci	.perform_reset		= sienna_cichlid_mode2_perform_reset,
27962306a36Sopenharmony_ci	.restore_hwcontext	= sienna_cichlid_mode2_restore_hwcontext,
28062306a36Sopenharmony_ci	.restore_env		= NULL,
28162306a36Sopenharmony_ci	.do_reset		= sienna_cichlid_mode2_reset,
28262306a36Sopenharmony_ci};
28362306a36Sopenharmony_ci
28462306a36Sopenharmony_ciint sienna_cichlid_reset_init(struct amdgpu_device *adev)
28562306a36Sopenharmony_ci{
28662306a36Sopenharmony_ci	struct amdgpu_reset_control *reset_ctl;
28762306a36Sopenharmony_ci
28862306a36Sopenharmony_ci	reset_ctl = kzalloc(sizeof(*reset_ctl), GFP_KERNEL);
28962306a36Sopenharmony_ci	if (!reset_ctl)
29062306a36Sopenharmony_ci		return -ENOMEM;
29162306a36Sopenharmony_ci
29262306a36Sopenharmony_ci	reset_ctl->handle = adev;
29362306a36Sopenharmony_ci	reset_ctl->async_reset = sienna_cichlid_async_reset;
29462306a36Sopenharmony_ci	reset_ctl->active_reset = AMD_RESET_METHOD_NONE;
29562306a36Sopenharmony_ci	reset_ctl->get_reset_handler = sienna_cichlid_get_reset_handler;
29662306a36Sopenharmony_ci
29762306a36Sopenharmony_ci	INIT_LIST_HEAD(&reset_ctl->reset_handlers);
29862306a36Sopenharmony_ci	INIT_WORK(&reset_ctl->reset_work, reset_ctl->async_reset);
29962306a36Sopenharmony_ci	/* Only mode2 is handled through reset control now */
30062306a36Sopenharmony_ci	amdgpu_reset_add_handler(reset_ctl, &sienna_cichlid_mode2_handler);
30162306a36Sopenharmony_ci
30262306a36Sopenharmony_ci	adev->reset_cntl = reset_ctl;
30362306a36Sopenharmony_ci
30462306a36Sopenharmony_ci	return 0;
30562306a36Sopenharmony_ci}
30662306a36Sopenharmony_ci
30762306a36Sopenharmony_ciint sienna_cichlid_reset_fini(struct amdgpu_device *adev)
30862306a36Sopenharmony_ci{
30962306a36Sopenharmony_ci	kfree(adev->reset_cntl);
31062306a36Sopenharmony_ci	adev->reset_cntl = NULL;
31162306a36Sopenharmony_ci	return 0;
31262306a36Sopenharmony_ci}
313