/*
 * PVA driver for T194
 *
 * Copyright (c) 2016-2018, NVIDIA Corporation.  All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */


#include <linux/export.h>
#include <linux/module.h>
#include <linux/pm_runtime.h>
#include <linux/fs.h>
#include <linux/of.h>
#include <linux/of_graph.h>
#include <linux/of_device.h>
#include <linux/of_platform.h>
#include <linux/platform_device.h>
#include <linux/uaccess.h>
#include <linux/delay.h>
#include <linux/slab.h>
#include <linux/firmware.h>
#include <linux/iommu.h>
#include <linux/reset.h>
#include <linux/platform/tegra/common.h>
#include <soc/tegra/chip-id.h>
#include <soc/tegra/fuse.h>

#include "nvhost_syncpt_unit_interface.h"
#include "dev.h"
#include "bus_client.h"
#include "nvhost_acm.h"
#include "t194/t194.h"
#include "nvhost_queue.h"
#include "pva_queue.h"
#include "pva.h"
#include "pva_regs.h"

#include "class_ids_t194.h"

/* Map PVA-A and PVA-B to respective configuration items in nvhost */
static struct of_device_id tegra_pva_of_match[] = {
	{
		.name = "pva0",
		.compatible = "nvidia,tegra194-pva",
		.data = (struct nvhost_device_data *)&t19_pva0_info },
	{
		.name = "pva1",
		.compatible = "nvidia,tegra194-pva",
		.data = (struct nvhost_device_data *)&t19_pva1_info },
	{ },
};

#define EVP_REG_NUM 8
static u32 pva_get_evp_reg(u32 index)
{
	u32 evp_reg[EVP_REG_NUM] = {
		evp_reset_addr_r(),
		evp_undef_addr_r(),
		evp_swi_addr_r(),
		evp_prefetch_abort_addr_r(),
		evp_data_abort_addr_r(),
		evp_rsvd_addr_r(),
		evp_irq_addr_r(),
		evp_fiq_addr_r()
	};

	return evp_reg[index];
}

/* Default buffer size (256 kbytes) used for ucode trace log*/
#define PVA_PRIV2_TRACE_LOG_BUFFER_SIZE 0x40000

#define R5_USER_SEGREG_OFFSET 0x40000000
static int pva_init_fw(struct platform_device *pdev)
{
	struct nvhost_device_data *pdata = platform_get_drvdata(pdev);
	struct pva *pva = pdata->private_data;
	struct pva_fw *fw_info = &pva->fw_info;
	struct pva_dma_alloc_info *priv1_buffer;
	struct pva_dma_alloc_info *priv2_buffer;
	u32 *ucode_ptr;
	int err = 0, w;
	u64 ucode_useg_addr;
	int sema_value = 0;

	nvhost_dbg_fn("");

	priv1_buffer = &fw_info->priv1_buffer;
	priv2_buffer = &fw_info->priv2_buffer;
	ucode_ptr = priv1_buffer->va;

	/* Set the Ucode Header address for R5 */
	/* Program user seg subtracting the offset */
	ucode_useg_addr = priv1_buffer->pa - R5_USER_SEGREG_OFFSET;
	host1x_writel(pdev, cfg_r5user_lsegreg_r(),
		PVA_LOW32(ucode_useg_addr));
	host1x_writel(pdev, cfg_r5user_usegreg_r(),
		PVA_EXTRACT64(ucode_useg_addr, 39, 32, u32));

	/* Program the extra memory to be used by R5 */
	ucode_useg_addr = priv2_buffer->pa - fw_info->priv2_reg_offset;
	host1x_writel(pdev, cfg_priv_ar2_start_r(), fw_info->priv2_reg_offset);
	host1x_writel(pdev, cfg_priv_ar2_end_r(),
			fw_info->priv2_reg_offset + priv2_buffer->size);
	host1x_writel(pdev, cfg_priv_ar2_lsegreg_r(),
		PVA_LOW32(ucode_useg_addr));
	host1x_writel(pdev, cfg_priv_ar2_usegreg_r(),
		PVA_EXTRACT64(ucode_useg_addr, 39, 32, u32));

	/* check the type of segments and their offset and address */
	for (w = 0; w < fw_info->hdr->nsegments; w++) {
		struct pva_ucode_seg *useg = (struct pva_ucode_seg *)
			((void *)ucode_ptr + PVA_UCODE_SEG_HDR_LENGTH
				+ (PVA_UCODE_SEG_HDR_LENGTH * w));

		switch (useg->type) {

		case PVA_UCODE_SEG_EVP: {
			/* First 32 bytes of the EVP payload are zeros.
			 * so skip first 32 bytes
			 */
			u32 *evpmem =
				(u32 *)((u8 *)ucode_ptr + useg->offset + 32);
			u32 i;

			for (i = 0; i < EVP_REG_NUM; i++)
				host1x_writel(pdev, pva_get_evp_reg(i),
					      evpmem[i]);
			break;
		}

		case PVA_UCODE_SEG_R5: {
			/* Subracting PRIV1 start for R5PRIV1 address */
			const u64 seg_addr = priv1_buffer->pa - useg->addr;
			/* Calculate segment start address */
			const u64 useg_addr = seg_addr + useg->offset;
			const u32 useg_addr_low = PVA_LOW32(useg_addr);
			const u32 useg_addr_high =
				PVA_EXTRACT64((useg_addr), 39, 32, u32);
			/* Calculate ar1 base and limit */
			const u32 ar1_start = useg->addr;
			const u32 ar1_end =
				useg->addr + priv1_buffer->size - useg->offset;

			host1x_writel(pdev, cfg_priv_ar1_start_r(), ar1_start);
			host1x_writel(pdev, cfg_priv_ar1_end_r(), ar1_end);
			host1x_writel(pdev, cfg_priv_ar1_lsegreg_r(),
				      useg_addr_low);
			host1x_writel(pdev, cfg_priv_ar1_usegreg_r(),
				      useg_addr_high);

			break;
		}

		}
	}

	/* Indicate the OS is waiting for PVA ready Interrupt */
	pva->mailbox_status = PVA_MBOX_STATUS_WFI;

	if (pva->r5_dbg_wait) {
		sema_value = PVA_WAIT_DEBUG;
		pva->timeout_enabled = false;
	}

	if (pva->slcg_disable)
		sema_value |= PVA_CG_DISABLE;

	if (pva->vmem_war_disable)
		sema_value |= PVA_VMEM_RD_WAR_DISABLE;

	sema_value |= (PVA_BOOT_INT | PVA_TEST_WAIT | PVA_VMEM_MBX_WAR_ENABLE);
	host1x_writel(pdev, hsp_ss0_set_r(), sema_value);

	/* Take R5 out of reset */
	host1x_writel(pdev, proc_cpuhalt_r(),
		proc_cpuhalt_ncpuhalt_f(proc_cpuhalt_ncpuhalt_done_v()));

	nvhost_dbg_fn("Waiting for PVA to be READY");

	/* Wait PVA to report itself as ready */
	err = pva_mailbox_wait_event(pva, 60000);
	if (err)
		goto wait_timeout;

	pva->mailbox_status = PVA_MBOX_STATUS_INVALID;

	nvhost_dbg_fn("PVA boot returned: %d", err);

	/* Check the ucode is with testmode enabled */
	if ((host1x_readl(pdev, hsp_ss0_state_r()) & PVA_TEST_MODE))
		err = pva_run_ucode_selftest(pdev);

wait_timeout:
	return err;
}

static int pva_free_fw(struct platform_device *pdev, struct pva *pva)
{
	struct pva_fw *fw_info = &pva->fw_info;

	if (pva->priv1_dma.va)
		dma_free_attrs(&pdev->dev, pva->priv1_dma.size,
		pva->priv1_dma.va, pva->priv1_dma.pa,
		__DMA_ATTR(fw_info->attrs));

	if (pva->priv2_dma.va)
		dma_free_attrs(&pdev->dev, pva->priv2_dma.size,
		pva->priv2_dma.va, pva->priv2_dma.pa,
		__DMA_ATTR(fw_info->attrs));

	memset(fw_info, 0, sizeof(struct pva_fw));

	return 0;
}

/*
 * NO IOMMU set 0x60000000 as start address.
 * With IOMMU set 0x80000000(>2GB) as startaddress
 */
#define DRAM_PVA_IOVA_START_ADDRESS 0x80000000
#define DRAM_PVA_NO_IOMMU_START_ADDRESS 0x60000000

static int pva_read_ucode(struct platform_device *pdev,
		const char *fw_name, struct pva *pva)
{
	int err = 0, w;
	u32 *ucode_ptr;
	const struct firmware *ucode_fw;
	struct pva_fw *fw_info = &pva->fw_info;
	struct pva_trace_log *trace = &pva->pva_trace;

	nvhost_dbg_fn("");

#if LINUX_VERSION_CODE < KERNEL_VERSION(4, 9, 0)
	init_dma_attrs(&fw_info->attrs);
#else
	fw_info->attrs = 0;
#endif

	ucode_fw = nvhost_client_request_firmware(pdev, fw_name);
	if (!ucode_fw) {
		nvhost_dbg_fn("pva firmware request failed");
		dev_err(&pdev->dev,
			"Failed to load the %s firmware\n", fw_name);
		err = -ENOENT;
		return err;
	}

	/* set to default size, will add support to modify through debugfs */
	fw_info->trace_buffer_size = PVA_PRIV2_TRACE_LOG_BUFFER_SIZE;

	fw_info->priv1_buffer.size = ucode_fw->size;

	/* Make sure the address is aligned to 4K */
	pva->priv1_dma.size = ALIGN(fw_info->priv1_buffer.size + SZ_4K, SZ_4K);

	/* Allocate memory to R5 for app code, data or to log information */
	pva->priv1_dma.va = dma_alloc_attrs(&pdev->dev,
				pva->priv1_dma.size,
				&pva->priv1_dma.pa,
				GFP_KERNEL, __DMA_ATTR(fw_info->attrs));

	if (!pva->priv1_dma.va) {
		err = -ENOMEM;
		goto clean_up;
	}

	/* Make sure the buffer allocated to R5 are 4K aligned */
	fw_info->priv1_buffer.va =
			(void *)ALIGN((u64)pva->priv1_dma.va, SZ_4K);
	fw_info->priv1_buffer.pa =
			(dma_addr_t)ALIGN((u64)pva->priv1_dma.pa, SZ_4K);

	ucode_ptr = fw_info->priv1_buffer.va;

	/* copy the whole thing taking into account endianness */
	for (w = 0; w < ucode_fw->size/sizeof(u32); w++)
		ucode_ptr[w] = le32_to_cpu(((__le32 *)ucode_fw->data)[w]);

	/* set the header location accordingly */
	fw_info->hdr = (struct pva_ucode_hdr *)ucode_ptr;

	/* check for the magic number  and header version*/
	if ((fw_info->hdr->magic != PVA_HDR_MAGIC) &&
		(fw_info->hdr->hdr_version != PVA_HDR_VERSION)) {
		dev_err(&pdev->dev, "Wrong PVA uCode header magic/version\n");
		err = -EINVAL;
	}

	/* find the size needed for priv2 buffer allocation */
	/* check the type of segments and their offset and address */
	for (w = 0; w < fw_info->hdr->nsegments; w++) {
		struct pva_ucode_seg *useg = (struct pva_ucode_seg *)
			((void *)ucode_ptr + PVA_UCODE_SEG_HDR_LENGTH
				+ (PVA_UCODE_SEG_HDR_LENGTH * w));

		switch (useg->type) {
		case PVA_UCODE_SEG_DRAM_CACHED:
			/* Total 2GB of contiguous memory for cache
			 * Set the DRAM CACHE physical addr as iova start
			 */
			if (pdev->dev.archdata.iommu)
				useg->phys_addr = DRAM_PVA_IOVA_START_ADDRESS;
			else
				useg->phys_addr = DRAM_PVA_NO_IOMMU_START_ADDRESS;
			break;
		case PVA_UCODE_SEG_DRAM_UNCACHED:
			/* Set the Uncache size as Zero */
			useg->size = 0;
			break;
		case PVA_UCODE_SEG_R5_OVERLAY:
		case PVA_UCODE_SEG_R5_CRASHDUMP:
		case PVA_UCODE_SEG_VPU_CRASHDUMP:
			fw_info->priv2_buffer.size += useg->size;
			break;
		case PVA_UCODE_SEG_TRACE_LOG:
			/* set the trace log buffer offset from priv2 start
			 * offset must be 64bytes aligned for dma usage
			 */
			fw_info->priv2_buffer.size =
				 ALIGN(fw_info->priv2_buffer.size + 64, 64);

			/* set the trace log buffer offset from priv2 start */
			useg->offset = fw_info->priv2_buffer.size;
			/* set os specified size if uCode passes zero size */
			if (!useg->size)
				useg->size = fw_info->trace_buffer_size;

			useg->addr = ALIGN(useg->addr + 64, 64);
			trace->size = useg->size;
			trace->offset = useg->offset;

			fw_info->priv2_buffer.size += useg->size;
			break;
		}
	}

	/* Make sure the address is aligned to 4K */
	pva->priv2_dma.size = ALIGN(fw_info->priv2_buffer.size + SZ_4K, SZ_4K);

	/* Allocate memory to R5 for app code, data or to log information */
	pva->priv2_dma.va = dma_alloc_attrs(&pdev->dev,
				pva->priv2_dma.size,
				&pva->priv2_dma.pa,
				GFP_KERNEL, __DMA_ATTR(fw_info->attrs));

	if (!pva->priv2_dma.va) {
		err = -ENOMEM;
		goto clean_up;
	}

	/* Make sure the buffer allocated to R5 are 4K aligned */
	fw_info->priv2_buffer.va =
			(void *)ALIGN((u64)pva->priv2_dma.va, SZ_4K);

	trace->addr = (void *)((u8 *)fw_info->priv2_buffer.va +
					trace->offset);
	memset(trace->addr, 0, trace->size);

	fw_info->priv2_buffer.pa =
			(dma_addr_t)ALIGN((u64)pva->priv2_dma.pa, SZ_4K);

	/* set the crashdump offsets and addresses */
	for (w = 0; w < fw_info->hdr->nsegments; w++) {
		struct pva_seg_info *seg_info = NULL;
		struct pva_ucode_seg *useg = (struct pva_ucode_seg *)
			((void *)ucode_ptr + PVA_UCODE_SEG_HDR_LENGTH
				+ (PVA_UCODE_SEG_HDR_LENGTH * w));
		int offset = useg->addr - fw_info->priv2_reg_offset;

		switch (useg->type) {
		case PVA_UCODE_SEG_R5_OVERLAY:
			fw_info->priv2_reg_offset = useg->addr;
			break;
		case PVA_UCODE_SEG_R5_CRASHDUMP:
			seg_info = &pva->debugfs_entry_r5.seg_info;
			break;
		case PVA_UCODE_SEG_VPU_CRASHDUMP:
			if (useg->id == 0)
				seg_info = &pva->debugfs_entry_vpu0.seg_info;
			else
				seg_info = &pva->debugfs_entry_vpu1.seg_info;
			break;
		}

		if (seg_info) {
			seg_info->offset = offset;
			seg_info->size = useg->size;
			seg_info->addr =
				(void *)((u8 *)fw_info->priv2_buffer.va +
								offset);
		}

	}
clean_up:
	release_firmware(ucode_fw);
	return err;
}

static int pva_load_fw(struct platform_device *pdev)
{
	int err = 0;
	struct nvhost_device_data *pdata = platform_get_drvdata(pdev);
	struct pva *pva = pdata->private_data;

	err = pva_read_ucode(pdev, pdata->firmware_name, pva);
	if (err < 0)
		goto load_fw_err;

	return err;

load_fw_err:
	pva_free_fw(pdev, pva);
	return err;
}

static int pva_alloc_vpu_function_table(struct pva *pva,
					struct pva_func_table *fn_table)
{
	uint32_t flags = PVA_CMD_INT_ON_ERR | PVA_CMD_INT_ON_COMPLETE;
	struct pva_mailbox_status_regs status;
	dma_addr_t dma_handle;
	uint32_t table_size;
	struct pva_cmd cmd;
	uint32_t entries;
	void *va;
	int err = 0;
	u32 nregs;

	nregs = pva_cmd_get_vpu_func_table(&cmd, 0, 0, flags);

	err = pva_mailbox_send_cmd_sync(pva, &cmd, nregs, &status);
	if (err < 0) {
		nvhost_warn(&pva->pdev->dev,
			"mbox function table cmd failed: %d\n", err);
		goto end;
	}

	table_size = status.status[PVA_CCQ_STATUS4_INDEX];
	entries = status.status[PVA_CCQ_STATUS5_INDEX];

	va = dma_alloc_coherent(&pva->pdev->dev, table_size,
				&dma_handle, GFP_KERNEL);
	if (va == NULL) {
		nvhost_warn(&pva->pdev->dev, "Unable to allocate the virtual address\n");
		err =  -ENOMEM;
		goto end;
	}

	fn_table->addr = va;
	fn_table->size = table_size;
	fn_table->handle = dma_handle;
	fn_table->entries = entries;

end:
	return err;
}

static int pva_get_vpu_function_table(struct pva *pva,
					struct pva_func_table *fn_table)
{
	uint32_t flags = PVA_CMD_INT_ON_ERR | PVA_CMD_INT_ON_COMPLETE;
	struct pva_mailbox_status_regs status;
	dma_addr_t dma_handle;
	uint32_t table_size;
	struct pva_cmd cmd;
	int err = 0;
	u32 nregs;

	dma_handle = fn_table->handle;
	table_size = fn_table->size;

	nregs = pva_cmd_get_vpu_func_table(&cmd, table_size, dma_handle, flags);

	/* Submit request to PVA and wait for response */
	err = pva_mailbox_send_cmd_sync(pva, &cmd, nregs, &status);
	if (err < 0)
		nvhost_warn(&pva->pdev->dev,
			"mbox function table cmd failed: %d\n", err);

	return err;

}

int pva_get_firmware_version(struct pva *pva,
			     struct pva_version_info *info)
{
	uint32_t flags = PVA_CMD_INT_ON_ERR | PVA_CMD_INT_ON_COMPLETE;
	struct pva_mailbox_status_regs status;
	struct pva_cmd cmd;
	int err = 0;
	u32 nregs;

	nregs = pva_cmd_R5_version(&cmd, flags);

	/* Submit request to PVA and wait for response */
	err = pva_mailbox_send_cmd_sync(pva, &cmd, nregs, &status);
	if (err < 0) {
		nvhost_warn(&pva->pdev->dev,
			"mbox get firmware version cmd failed: %d\n", err);

		return err;
	}

	info->pva_r5_version = status.status[PVA_CCQ_STATUS4_INDEX];
	info->pva_compat_version = status.status[PVA_CCQ_STATUS5_INDEX];
	info->pva_revision = status.status[PVA_CCQ_STATUS6_INDEX];
	info->pva_built_on = status.status[PVA_CCQ_STATUS7_INDEX];

	return err;
}

void pva_dealloc_vpu_function_table(struct pva *pva,
					struct pva_func_table *fn_table)
{
	dma_addr_t dma_handle = fn_table->handle;
	uint32_t table_size = fn_table->size;
	void *va = fn_table->addr;

	if (va)
		dma_free_coherent(&pva->pdev->dev,
				table_size, va, dma_handle);
}

int pva_alloc_and_populate_function_table(struct pva *pva,
					struct pva_func_table *fn_table)
{
	int ret = 0;

	ret = pva_alloc_vpu_function_table(pva, fn_table);
	if (ret < 0)
		goto err_alloc_vpu_function_table;

	ret = pva_get_vpu_function_table(pva, fn_table);
	if (ret < 0)
		goto err_get_vpu_function_table;

	return 0;

err_get_vpu_function_table:
	pva_dealloc_vpu_function_table(pva, fn_table);
err_alloc_vpu_function_table:
	return ret;

}

int pva_set_log_level(struct pva *pva,
			     u32 log_level)
{
	uint32_t flags = PVA_CMD_INT_ON_ERR | PVA_CMD_INT_ON_COMPLETE;
	struct pva_mailbox_status_regs status;
	struct pva_cmd cmd;
	int err = 0;
	u32 nregs;

	nregs = pva_cmd_set_logging_level(&cmd, log_level, flags);

	err = pva_mailbox_send_cmd_sync(pva, &cmd, nregs, &status);
	if (err < 0)
		nvhost_warn(&pva->pdev->dev,
			"mbox set log level failed: %d\n", err);

	return err;
}

static void pva_restore_attributes(struct pva * pva)
{
	struct nvhost_queue_pool *pool = pva->pool;
	struct pva_queue_set_attribute set_attr;
	struct pva_queue_attribute *attrs;
	unsigned int i = 0;
	int err;

	set_attr.pva = pva;
	set_attr.bootup = true;

	mutex_lock(&pool->queue_lock);
	for_each_set_bit_from(i, &pool->alloc_table, pool->max_queue_cnt) {
		u32 id;

		mutex_lock(&pool->queues[i].attr_lock);
		attrs = pool->queues[i].attr;
		if (attrs == NULL) {
			mutex_unlock(&pool->queues[i].attr_lock);
			continue;
		}

		for (id = 1; id < QUEUE_ATTR_MAX; id++) {
			set_attr.attr = &attrs[id];
			err = nvhost_queue_set_attr(&pool->queues[i],
						    &set_attr);
			if (err) {
				dev_err(&pva->pdev->dev,
					"unable to set attribute %u to queue %u\n",
					id, i);
			}
		}

		mutex_unlock(&pool->queues[i].attr_lock);
	}
	mutex_unlock(&pool->queue_lock);
}

int pva_finalize_poweron(struct platform_device *pdev)
{
	struct nvhost_device_data *pdata = platform_get_drvdata(pdev);
	struct pva *pva = pdata->private_data;
	int err = 0;

	/* Enable LIC_INTERRUPT line for HSP1 and WDT */
	host1x_writel(pva->pdev, sec_lic_intr_enable_r(),
		sec_lic_intr_enable_hsp_f(SEC_LIC_INTR_HSP1) |
		sec_lic_intr_enable_wdt_f(SEC_LIC_INTR_WDT));

	err = pva_load_fw(pdev);
	if (err < 0)
		goto err_poweron;

	enable_irq(pva->irq);

	err = pva_init_fw(pdev);
	if (err < 0)
		goto err_poweron;

	pva_set_log_level(pva, pva->log_level);

	/* Restore the attributes */
	pva_restore_attributes(pva);

	pva->booted = true;

	return err;

err_poweron:
	disable_irq(pva->irq);
	return err;
}

int pva_prepare_poweroff(struct platform_device *pdev)
{
	struct nvhost_device_data *pdata = platform_get_drvdata(pdev);
	struct pva *pva = pdata->private_data;

	/*
	 * Disable IRQs. Interrupt handler won't be under execution after the
	 * call returns.
	 */
	disable_irq(pva->irq);

	/* Put PVA to reset to ensure that the firmware doesn't get accessed */
	reset_control_assert(pdata->reset_control);
	pva->booted = false;

	pva_free_fw(pdev, pva);

	return 0;
}

static int pva_probe(struct platform_device *pdev)
{
	struct device *dev = &pdev->dev;
	struct nvhost_device_data *pdata;
	const struct of_device_id *match;
	struct pva *pva;
	int err = 0;

	nvhost_dbg_fn("%s", __func__);

	match = of_match_device(tegra_pva_of_match, dev);
	pdata = (struct nvhost_device_data *)match->data;

	WARN_ON(!pdata);
	if (!pdata) {
		dev_info(dev, "no platform data\n");
		err = -ENODATA;
		goto err_get_pdata;
	}

	if (tegra_get_chipid() == TEGRA_CHIPID_TEGRA19 &&
		tegra_get_sku_id() == 0x9E) {
		dev_err(dev, "PVA IP is disabled in SKU\n");
		err = -ENODEV;
		goto err_no_ip;
	}

	if (tegra_get_chipid() == TEGRA_CHIPID_TEGRA19 &&
	    tegra_get_sku_id() == 0x9F &&
	    pdata->class == NV_PVA1_CLASS_ID) {
		dev_err(dev, "PVA1 IP is disabled in SKU\n");
		err = -ENODEV;
		goto err_no_ip;
	}

	pva = devm_kzalloc(dev, sizeof(*pva), GFP_KERNEL);
	if (!pva) {
		err = -ENOMEM;
		goto err_alloc_pva;
	}

	/* Initialize PVA private data */
	pva->pdev = pdev;


	/* Enable powergating and timeout only on silicon */
	if (!tegra_platform_is_silicon()) {
		pdata->can_powergate = false;
		pva->timeout_enabled = false;
	} else {
		pva->timeout_enabled = true;
	}


	/* Initialize nvhost specific data */
	pdata->pdev = pdev;
	mutex_init(&pdata->lock);
	pdata->private_data = pva;
	platform_set_drvdata(pdev, pdata);
	init_waitqueue_head(&pva->mailbox_waitqueue);
	mutex_init(&pva->mailbox_mutex);
	mutex_init(&pva->ccq_mutex);
	pva->submit_mode = PVA_SUBMIT_MODE_MMIO_CCQ;
	pva->slcg_disable = 0;
	pva->vmem_war_disable = 0;
	pva->vpu_perf_counters_enable = false;

#ifdef __linux__
#if LINUX_VERSION_CODE >= KERNEL_VERSION(4, 9, 0)
	if (tegra_chip_get_revision() != TEGRA194_REVISION_A01)
		pva->vmem_war_disable = 1;
#endif
#endif

	/* Map MMIO range to kernel space */
	err = nvhost_client_device_get_resources(pdev);
	if (err < 0)
		goto err_get_resources;

	/* Get clocks */
	err = nvhost_module_init(pdev);
	if (err < 0)
		goto err_module_init;

	/*
	 * Add this to nvhost device list, initialize scaling,
	 * setup memory management for the device, create dev nodes
	 */
	err = nvhost_client_device_init(pdev);
	if (err < 0)
		goto err_client_device_init;

	pva->pool = nvhost_queue_init(pdev, &pva_queue_ops,
					MAX_PVA_QUEUE_COUNT);
	if (IS_ERR(pva->pool)) {
		err = PTR_ERR(pva->pool);
		goto err_queue_init;
	}

	err = pva_register_isr(pdev);
	if (err < 0)
		goto err_isr_init;

	pva_abort_init(pva);

	err = nvhost_syncpt_unit_interface_init(pdev);
	if (err)
		goto err_mss_init;

#ifdef CONFIG_DEBUG_FS
	pva_debugfs_init(pdev);
#endif
	return 0;

err_mss_init:
err_isr_init:
	nvhost_queue_deinit(pva->pool);
err_queue_init:
	nvhost_client_device_release(pdev);
err_client_device_init:
	nvhost_module_deinit(pdev);
err_module_init:
err_get_resources:
	devm_kfree(dev, pva);
err_alloc_pva:
err_no_ip:
err_get_pdata:

	return err;
}

static int __exit pva_remove(struct platform_device *pdev)
{
	struct nvhost_device_data *pdata = platform_get_drvdata(pdev);
	struct pva *pva = pdata->private_data;

	nvhost_queue_deinit(pva->pool);
	nvhost_client_device_release(pdev);
	free_irq(pva->irq, pdata);

	return 0;
}

static struct platform_driver pva_driver = {
	.probe = pva_probe,
	.remove = __exit_p(pva_remove),
	.driver = {
		.owner = THIS_MODULE,
		.name = "pva",
#ifdef CONFIG_OF
		.of_match_table = tegra_pva_of_match,
#endif
#ifdef CONFIG_PM
		.pm = &nvhost_module_pm_ops,
#endif
	},
};

module_platform_driver(pva_driver);