xref: /OK3568_Linux_fs/kernel/drivers/video/rockchip/rve/rve_drv.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun  * Copyright (C) Rockchip Electronics Co., Ltd.
4*4882a593Smuzhiyun  *
5*4882a593Smuzhiyun  * Author: Huang Lee <Putin.li@rock-chips.com>
6*4882a593Smuzhiyun  */
7*4882a593Smuzhiyun 
8*4882a593Smuzhiyun #define pr_fmt(fmt) "rve: " fmt
9*4882a593Smuzhiyun 
10*4882a593Smuzhiyun #include "rve_job.h"
11*4882a593Smuzhiyun #include "rve_fence.h"
12*4882a593Smuzhiyun #include "rve_debugger.h"
13*4882a593Smuzhiyun #include "rve_reg.h"
14*4882a593Smuzhiyun 
15*4882a593Smuzhiyun struct rve_drvdata_t *rve_drvdata;
16*4882a593Smuzhiyun 
17*4882a593Smuzhiyun /* set hrtimer */
18*4882a593Smuzhiyun static struct hrtimer timer;
19*4882a593Smuzhiyun static ktime_t kt;
20*4882a593Smuzhiyun 
21*4882a593Smuzhiyun static const struct rve_backend_ops rve_ops = {
22*4882a593Smuzhiyun 	.get_version = rve_get_version,
23*4882a593Smuzhiyun 	.set_reg = rve_set_reg,
24*4882a593Smuzhiyun 	.init_reg = rve_init_reg,
25*4882a593Smuzhiyun 	.soft_reset = rve_soft_reset
26*4882a593Smuzhiyun };
27*4882a593Smuzhiyun 
rve_ctx_set_debuf_info_cb(int id,void * ptr,void * data)28*4882a593Smuzhiyun static int rve_ctx_set_debuf_info_cb(int id, void *ptr, void *data)
29*4882a593Smuzhiyun {
30*4882a593Smuzhiyun 	struct rve_internal_ctx_t *ctx = ptr;
31*4882a593Smuzhiyun 	unsigned long flags;
32*4882a593Smuzhiyun 
33*4882a593Smuzhiyun 	spin_lock_irqsave(&ctx->lock, flags);
34*4882a593Smuzhiyun 
35*4882a593Smuzhiyun 	ctx->debug_info.max_cost_time_per_sec = 0;
36*4882a593Smuzhiyun 	ctx->debug_info.hw_time_total = 0;
37*4882a593Smuzhiyun 
38*4882a593Smuzhiyun 	spin_unlock_irqrestore(&ctx->lock, flags);
39*4882a593Smuzhiyun 
40*4882a593Smuzhiyun 	return 0;
41*4882a593Smuzhiyun }
42*4882a593Smuzhiyun 
hrtimer_handler(struct hrtimer * timer)43*4882a593Smuzhiyun static enum hrtimer_restart hrtimer_handler(struct hrtimer *timer)
44*4882a593Smuzhiyun {
45*4882a593Smuzhiyun 	struct rve_drvdata_t *rve = rve_drvdata;
46*4882a593Smuzhiyun 	struct rve_scheduler_t *scheduler = NULL;
47*4882a593Smuzhiyun 	struct rve_pending_ctx_manager *ctx_manager;
48*4882a593Smuzhiyun 	struct rve_job *job = NULL;
49*4882a593Smuzhiyun 	unsigned long flags;
50*4882a593Smuzhiyun 	int i;
51*4882a593Smuzhiyun 
52*4882a593Smuzhiyun 	ktime_t now = ktime_get();
53*4882a593Smuzhiyun 
54*4882a593Smuzhiyun 	for (i = 0; i < rve->num_of_scheduler; i++) {
55*4882a593Smuzhiyun 		scheduler = rve->scheduler[i];
56*4882a593Smuzhiyun 
57*4882a593Smuzhiyun 		spin_lock_irqsave(&scheduler->irq_lock, flags);
58*4882a593Smuzhiyun 
59*4882a593Smuzhiyun 		/* if timer action on job running */
60*4882a593Smuzhiyun 		job = scheduler->running_job;
61*4882a593Smuzhiyun 		if (job) {
62*4882a593Smuzhiyun 			scheduler->timer.busy_time += ktime_us_delta(now, job->hw_recoder_time);
63*4882a593Smuzhiyun 			job->hw_recoder_time = now;
64*4882a593Smuzhiyun 		}
65*4882a593Smuzhiyun 
66*4882a593Smuzhiyun 		scheduler->timer.busy_time_record = scheduler->timer.busy_time;
67*4882a593Smuzhiyun 		scheduler->timer.busy_time = 0;
68*4882a593Smuzhiyun 
69*4882a593Smuzhiyun 		/* monitor */
70*4882a593Smuzhiyun 		scheduler->session.rd_bandwidth = 0;
71*4882a593Smuzhiyun 		scheduler->session.wr_bandwidth = 0;
72*4882a593Smuzhiyun 		scheduler->session.cycle_cnt = 0;
73*4882a593Smuzhiyun 
74*4882a593Smuzhiyun 		for (i = 0; i < RVE_MAX_PID_INFO; i++) {
75*4882a593Smuzhiyun 			if (scheduler->session.pid_info[i].pid > 0)
76*4882a593Smuzhiyun 				scheduler->session.pid_info[i].hw_time_total = 0;
77*4882a593Smuzhiyun 		}
78*4882a593Smuzhiyun 
79*4882a593Smuzhiyun 		spin_unlock_irqrestore(&scheduler->irq_lock, flags);
80*4882a593Smuzhiyun 
81*4882a593Smuzhiyun 		ctx_manager = rve_drvdata->pend_ctx_manager;
82*4882a593Smuzhiyun 
83*4882a593Smuzhiyun 		spin_lock_irqsave(&ctx_manager->lock, flags);
84*4882a593Smuzhiyun 
85*4882a593Smuzhiyun 		idr_for_each(&ctx_manager->ctx_id_idr, &rve_ctx_set_debuf_info_cb, ctx_manager);
86*4882a593Smuzhiyun 
87*4882a593Smuzhiyun 		spin_unlock_irqrestore(&ctx_manager->lock, flags);
88*4882a593Smuzhiyun 	}
89*4882a593Smuzhiyun 
90*4882a593Smuzhiyun 	hrtimer_forward_now(timer, kt);
91*4882a593Smuzhiyun 	return HRTIMER_RESTART;
92*4882a593Smuzhiyun }
93*4882a593Smuzhiyun 
rve_init_timer(void)94*4882a593Smuzhiyun static void rve_init_timer(void)
95*4882a593Smuzhiyun {
96*4882a593Smuzhiyun 	kt = ktime_set(0, RVE_LOAD_INTERVAL);
97*4882a593Smuzhiyun 
98*4882a593Smuzhiyun 	hrtimer_init(&timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
99*4882a593Smuzhiyun 
100*4882a593Smuzhiyun 	timer.function = hrtimer_handler;
101*4882a593Smuzhiyun 
102*4882a593Smuzhiyun 	hrtimer_start(&timer, kt, HRTIMER_MODE_REL);
103*4882a593Smuzhiyun }
104*4882a593Smuzhiyun 
rve_cancel_timer(void)105*4882a593Smuzhiyun static void rve_cancel_timer(void)
106*4882a593Smuzhiyun {
107*4882a593Smuzhiyun 	hrtimer_cancel(&timer);
108*4882a593Smuzhiyun }
109*4882a593Smuzhiyun 
110*4882a593Smuzhiyun #ifndef RVE_PD_AWAYS_ON
rve_power_enable(struct rve_scheduler_t * scheduler)111*4882a593Smuzhiyun int rve_power_enable(struct rve_scheduler_t *scheduler)
112*4882a593Smuzhiyun {
113*4882a593Smuzhiyun 	int ret = -EINVAL;
114*4882a593Smuzhiyun 	int i;
115*4882a593Smuzhiyun 
116*4882a593Smuzhiyun 	pm_runtime_get_sync(scheduler->dev);
117*4882a593Smuzhiyun 	pm_stay_awake(scheduler->dev);
118*4882a593Smuzhiyun 
119*4882a593Smuzhiyun 	for (i = 0; i < scheduler->num_clks; i++) {
120*4882a593Smuzhiyun 		if (!IS_ERR(scheduler->clks[i])) {
121*4882a593Smuzhiyun 			ret = clk_prepare_enable(scheduler->clks[i]);
122*4882a593Smuzhiyun 			if (ret < 0)
123*4882a593Smuzhiyun 				goto err_enable_clk;
124*4882a593Smuzhiyun 		}
125*4882a593Smuzhiyun 	}
126*4882a593Smuzhiyun 
127*4882a593Smuzhiyun 	scheduler->session.pd_refcount++;
128*4882a593Smuzhiyun 
129*4882a593Smuzhiyun 	return 0;
130*4882a593Smuzhiyun 
131*4882a593Smuzhiyun err_enable_clk:
132*4882a593Smuzhiyun 	for (--i; i >= 0; --i)
133*4882a593Smuzhiyun 		if (!IS_ERR(scheduler->clks[i]))
134*4882a593Smuzhiyun 			clk_disable_unprepare(scheduler->clks[i]);
135*4882a593Smuzhiyun 
136*4882a593Smuzhiyun 	pm_relax(scheduler->dev);
137*4882a593Smuzhiyun 	pm_runtime_put_sync_suspend(scheduler->dev);
138*4882a593Smuzhiyun 
139*4882a593Smuzhiyun 	return ret;
140*4882a593Smuzhiyun }
141*4882a593Smuzhiyun 
rve_power_disable(struct rve_scheduler_t * scheduler)142*4882a593Smuzhiyun int rve_power_disable(struct rve_scheduler_t *scheduler)
143*4882a593Smuzhiyun {
144*4882a593Smuzhiyun 	int i;
145*4882a593Smuzhiyun 
146*4882a593Smuzhiyun 	for (i = scheduler->num_clks - 1; i >= 0; i--)
147*4882a593Smuzhiyun 		if (!IS_ERR(scheduler->clks[i]))
148*4882a593Smuzhiyun 			clk_disable_unprepare(scheduler->clks[i]);
149*4882a593Smuzhiyun 
150*4882a593Smuzhiyun 	pm_relax(scheduler->dev);
151*4882a593Smuzhiyun 	pm_runtime_put_sync_suspend(scheduler->dev);
152*4882a593Smuzhiyun 
153*4882a593Smuzhiyun 	scheduler->session.pd_refcount--;
154*4882a593Smuzhiyun 
155*4882a593Smuzhiyun 	return 0;
156*4882a593Smuzhiyun }
157*4882a593Smuzhiyun 
158*4882a593Smuzhiyun #endif //RVE_PD_AWAYS_ON
159*4882a593Smuzhiyun 
rve_session_manager_init(struct rve_session_manager ** session_manager_ptr)160*4882a593Smuzhiyun static int rve_session_manager_init(struct rve_session_manager **session_manager_ptr)
161*4882a593Smuzhiyun {
162*4882a593Smuzhiyun 	struct rve_session_manager *session_manager = NULL;
163*4882a593Smuzhiyun 
164*4882a593Smuzhiyun 	*session_manager_ptr = kzalloc(sizeof(struct rve_session_manager), GFP_KERNEL);
165*4882a593Smuzhiyun 	if (*session_manager_ptr == NULL) {
166*4882a593Smuzhiyun 		pr_err("can not kzalloc for rve_session_manager\n");
167*4882a593Smuzhiyun 		return -ENOMEM;
168*4882a593Smuzhiyun 	}
169*4882a593Smuzhiyun 
170*4882a593Smuzhiyun 	session_manager = *session_manager_ptr;
171*4882a593Smuzhiyun 
172*4882a593Smuzhiyun 	mutex_init(&session_manager->lock);
173*4882a593Smuzhiyun 
174*4882a593Smuzhiyun 	idr_init_base(&session_manager->ctx_id_idr, 1);
175*4882a593Smuzhiyun 
176*4882a593Smuzhiyun 	return 0;
177*4882a593Smuzhiyun }
178*4882a593Smuzhiyun 
179*4882a593Smuzhiyun /*
180*4882a593Smuzhiyun  * Called at driver close to release the rve session's id references.
181*4882a593Smuzhiyun  */
rve_session_free_remove_idr_cb(int id,void * ptr,void * data)182*4882a593Smuzhiyun static int rve_session_free_remove_idr_cb(int id, void *ptr, void *data)
183*4882a593Smuzhiyun {
184*4882a593Smuzhiyun 	struct rve_session *session = ptr;
185*4882a593Smuzhiyun 
186*4882a593Smuzhiyun 	idr_remove(&rve_drvdata->session_manager->ctx_id_idr, session->id);
187*4882a593Smuzhiyun 	kfree(session);
188*4882a593Smuzhiyun 
189*4882a593Smuzhiyun 	return 0;
190*4882a593Smuzhiyun }
191*4882a593Smuzhiyun 
rve_session_free_remove_idr(struct rve_session * session)192*4882a593Smuzhiyun static int rve_session_free_remove_idr(struct rve_session *session)
193*4882a593Smuzhiyun {
194*4882a593Smuzhiyun 	struct rve_session_manager *session_manager;
195*4882a593Smuzhiyun 
196*4882a593Smuzhiyun 	session_manager = rve_drvdata->session_manager;
197*4882a593Smuzhiyun 
198*4882a593Smuzhiyun 	mutex_lock(&session_manager->lock);
199*4882a593Smuzhiyun 
200*4882a593Smuzhiyun 	session_manager->session_cnt--;
201*4882a593Smuzhiyun 	idr_remove(&session_manager->ctx_id_idr, session->id);
202*4882a593Smuzhiyun 
203*4882a593Smuzhiyun 	mutex_unlock(&session_manager->lock);
204*4882a593Smuzhiyun 
205*4882a593Smuzhiyun 	return 0;
206*4882a593Smuzhiyun }
207*4882a593Smuzhiyun 
rve_session_manager_remove(struct rve_session_manager ** session_manager_ptr)208*4882a593Smuzhiyun static int rve_session_manager_remove(struct rve_session_manager **session_manager_ptr)
209*4882a593Smuzhiyun {
210*4882a593Smuzhiyun 	struct rve_session_manager *session_manager = *session_manager_ptr;
211*4882a593Smuzhiyun 
212*4882a593Smuzhiyun 	mutex_lock(&session_manager->lock);
213*4882a593Smuzhiyun 
214*4882a593Smuzhiyun 	idr_for_each(&session_manager->ctx_id_idr, &rve_session_free_remove_idr_cb, session_manager);
215*4882a593Smuzhiyun 	idr_destroy(&session_manager->ctx_id_idr);
216*4882a593Smuzhiyun 
217*4882a593Smuzhiyun 	mutex_unlock(&session_manager->lock);
218*4882a593Smuzhiyun 
219*4882a593Smuzhiyun 	kfree(*session_manager_ptr);
220*4882a593Smuzhiyun 
221*4882a593Smuzhiyun 	*session_manager_ptr = NULL;
222*4882a593Smuzhiyun 
223*4882a593Smuzhiyun 	return 0;
224*4882a593Smuzhiyun }
225*4882a593Smuzhiyun 
rve_session_init(void)226*4882a593Smuzhiyun static struct rve_session *rve_session_init(void)
227*4882a593Smuzhiyun {
228*4882a593Smuzhiyun 	struct rve_session_manager *session_manager = NULL;
229*4882a593Smuzhiyun 	struct rve_session *session = kzalloc(sizeof(*session), GFP_KERNEL);
230*4882a593Smuzhiyun 
231*4882a593Smuzhiyun 	session_manager = rve_drvdata->session_manager;
232*4882a593Smuzhiyun 	if (session_manager == NULL) {
233*4882a593Smuzhiyun 		pr_err("rve_session_manager is null!\n");
234*4882a593Smuzhiyun 		kfree(session);
235*4882a593Smuzhiyun 		return NULL;
236*4882a593Smuzhiyun 	}
237*4882a593Smuzhiyun 
238*4882a593Smuzhiyun 	mutex_lock(&session_manager->lock);
239*4882a593Smuzhiyun 
240*4882a593Smuzhiyun 	idr_preload(GFP_KERNEL);
241*4882a593Smuzhiyun 	session->id = idr_alloc(&session_manager->ctx_id_idr, session, 1, 0, GFP_ATOMIC);
242*4882a593Smuzhiyun 	session_manager->session_cnt++;
243*4882a593Smuzhiyun 	idr_preload_end();
244*4882a593Smuzhiyun 
245*4882a593Smuzhiyun 	mutex_unlock(&session_manager->lock);
246*4882a593Smuzhiyun 
247*4882a593Smuzhiyun 	session->tgid = current->tgid;
248*4882a593Smuzhiyun 
249*4882a593Smuzhiyun 	return session;
250*4882a593Smuzhiyun }
251*4882a593Smuzhiyun 
rve_session_deinit(struct rve_session * session)252*4882a593Smuzhiyun static int rve_session_deinit(struct rve_session *session)
253*4882a593Smuzhiyun {
254*4882a593Smuzhiyun 	int ctx_id;
255*4882a593Smuzhiyun 	struct rve_pending_ctx_manager *ctx_manager;
256*4882a593Smuzhiyun 	struct rve_internal_ctx_t *ctx;
257*4882a593Smuzhiyun 	unsigned long flags;
258*4882a593Smuzhiyun 
259*4882a593Smuzhiyun 	ctx_manager = rve_drvdata->pend_ctx_manager;
260*4882a593Smuzhiyun 
261*4882a593Smuzhiyun 	spin_lock_irqsave(&ctx_manager->lock, flags);
262*4882a593Smuzhiyun 
263*4882a593Smuzhiyun 	idr_for_each_entry(&ctx_manager->ctx_id_idr, ctx, ctx_id) {
264*4882a593Smuzhiyun 
265*4882a593Smuzhiyun 		spin_unlock_irqrestore(&ctx_manager->lock, flags);
266*4882a593Smuzhiyun 
267*4882a593Smuzhiyun 		if (session == ctx->session)
268*4882a593Smuzhiyun 			kref_put(&ctx->refcount, rve_internal_ctx_kref_release);
269*4882a593Smuzhiyun 
270*4882a593Smuzhiyun 		spin_lock_irqsave(&ctx_manager->lock, flags);
271*4882a593Smuzhiyun 	}
272*4882a593Smuzhiyun 
273*4882a593Smuzhiyun 	spin_unlock_irqrestore(&ctx_manager->lock, flags);
274*4882a593Smuzhiyun 
275*4882a593Smuzhiyun 	rve_job_session_destroy(session);
276*4882a593Smuzhiyun 
277*4882a593Smuzhiyun 	rve_session_free_remove_idr(session);
278*4882a593Smuzhiyun 	kfree(session);
279*4882a593Smuzhiyun 
280*4882a593Smuzhiyun 	return 0;
281*4882a593Smuzhiyun }
282*4882a593Smuzhiyun 
rve_ioctl_cmd_start(unsigned long arg,struct rve_session * session)283*4882a593Smuzhiyun static long rve_ioctl_cmd_start(unsigned long arg, struct rve_session *session)
284*4882a593Smuzhiyun {
285*4882a593Smuzhiyun 	int rve_user_ctx_id;
286*4882a593Smuzhiyun 	int ret = 0;
287*4882a593Smuzhiyun 
288*4882a593Smuzhiyun 	rve_user_ctx_id = rve_internal_ctx_alloc_to_get_idr_id(session);
289*4882a593Smuzhiyun 
290*4882a593Smuzhiyun 	if (copy_to_user((void *)arg, &rve_user_ctx_id, sizeof(int)))
291*4882a593Smuzhiyun 		ret = -EFAULT;
292*4882a593Smuzhiyun 
293*4882a593Smuzhiyun 	return ret;
294*4882a593Smuzhiyun }
295*4882a593Smuzhiyun 
rve_ioctl_cmd_config(unsigned long arg)296*4882a593Smuzhiyun static long rve_ioctl_cmd_config(unsigned long arg)
297*4882a593Smuzhiyun {
298*4882a593Smuzhiyun 	struct rve_user_ctx_t user_ctx;
299*4882a593Smuzhiyun 	int ret = 0;
300*4882a593Smuzhiyun 
301*4882a593Smuzhiyun 	if (unlikely(copy_from_user(&user_ctx, (struct rve_user_ctx_t *)arg,
302*4882a593Smuzhiyun 			sizeof(user_ctx)))) {
303*4882a593Smuzhiyun 		pr_err("rve_user_ctx copy_from_user failed!\n");
304*4882a593Smuzhiyun 		return -EFAULT;
305*4882a593Smuzhiyun 	}
306*4882a593Smuzhiyun 
307*4882a593Smuzhiyun /* TODO:
308*4882a593Smuzhiyun  *	if (rve_user_ctx.cmd_num > RVE_CMD_NUM_MAX) {
309*4882a593Smuzhiyun  *		pr_err("Cannot import more than %d buffers at a time!\n",
310*4882a593Smuzhiyun  *			RVE_CMD_NUM_MAX);
311*4882a593Smuzhiyun  *		return -EFBIG;
312*4882a593Smuzhiyun  *	}
313*4882a593Smuzhiyun  */
314*4882a593Smuzhiyun 
315*4882a593Smuzhiyun 	if (user_ctx.id <= 0) {
316*4882a593Smuzhiyun 		pr_err("ctx id[%d] is invalid", user_ctx.id);
317*4882a593Smuzhiyun 		return -EINVAL;
318*4882a593Smuzhiyun 	}
319*4882a593Smuzhiyun 
320*4882a593Smuzhiyun 	if (DEBUGGER_EN(MSG))
321*4882a593Smuzhiyun 		pr_info("config cmd id = %d", user_ctx.id);
322*4882a593Smuzhiyun 
323*4882a593Smuzhiyun 	/* find internal_ctx to set cmd by user ctx (internal ctx id) */
324*4882a593Smuzhiyun 	ret = rve_job_config_by_user_ctx(&user_ctx);
325*4882a593Smuzhiyun 	if (ret < 0) {
326*4882a593Smuzhiyun 		pr_err("config ctx id[%d] failed!\n", user_ctx.id);
327*4882a593Smuzhiyun 		return -EFAULT;
328*4882a593Smuzhiyun 	}
329*4882a593Smuzhiyun 
330*4882a593Smuzhiyun 	return ret;
331*4882a593Smuzhiyun }
332*4882a593Smuzhiyun 
rve_ioctl_cmd_end(unsigned long arg)333*4882a593Smuzhiyun static long rve_ioctl_cmd_end(unsigned long arg)
334*4882a593Smuzhiyun {
335*4882a593Smuzhiyun 	struct rve_user_ctx_t rve_user_ctx;
336*4882a593Smuzhiyun 	int ret = 0;
337*4882a593Smuzhiyun 
338*4882a593Smuzhiyun 	if (unlikely(copy_from_user(&rve_user_ctx, (uint32_t *)arg,
339*4882a593Smuzhiyun 			sizeof(rve_user_ctx)))) {
340*4882a593Smuzhiyun 		pr_err("rve_user_ctx copy_from_user failed!\n");
341*4882a593Smuzhiyun 		return -EFAULT;
342*4882a593Smuzhiyun 	}
343*4882a593Smuzhiyun 
344*4882a593Smuzhiyun 	if (DEBUGGER_EN(MSG))
345*4882a593Smuzhiyun 		pr_info("config end id = %d", rve_user_ctx.id);
346*4882a593Smuzhiyun 
347*4882a593Smuzhiyun 	/* find internal_ctx to set cmd by user ctx (internal ctx id) */
348*4882a593Smuzhiyun 	ret = rve_job_commit_by_user_ctx(&rve_user_ctx);
349*4882a593Smuzhiyun 	if (ret < 0) {
350*4882a593Smuzhiyun 		pr_err("commit ctx id[%d] failed!\n", rve_user_ctx.id);
351*4882a593Smuzhiyun 		return -EFAULT;
352*4882a593Smuzhiyun 	}
353*4882a593Smuzhiyun 
354*4882a593Smuzhiyun 	if (copy_to_user((struct rve_user_ctx_t *)arg,
355*4882a593Smuzhiyun 			&rve_user_ctx, sizeof(struct rve_user_ctx_t))) {
356*4882a593Smuzhiyun 		pr_err("rve_user_ctx copy_to_user failed\n");
357*4882a593Smuzhiyun 		return -EFAULT;
358*4882a593Smuzhiyun 	}
359*4882a593Smuzhiyun 
360*4882a593Smuzhiyun 	return ret;
361*4882a593Smuzhiyun }
362*4882a593Smuzhiyun 
rve_ioctl_cmd_cancel(unsigned long arg)363*4882a593Smuzhiyun static long rve_ioctl_cmd_cancel(unsigned long arg)
364*4882a593Smuzhiyun {
365*4882a593Smuzhiyun 	uint32_t rve_user_ctx_id;
366*4882a593Smuzhiyun 	int ret = 0;
367*4882a593Smuzhiyun 
368*4882a593Smuzhiyun 	if (unlikely(copy_from_user(&rve_user_ctx_id, (uint32_t *)arg,
369*4882a593Smuzhiyun 			sizeof(uint32_t)))) {
370*4882a593Smuzhiyun 		pr_err("rve_user_ctx copy_from_user failed!\n");
371*4882a593Smuzhiyun 		return -EFAULT;
372*4882a593Smuzhiyun 	}
373*4882a593Smuzhiyun 
374*4882a593Smuzhiyun 	if (DEBUGGER_EN(MSG))
375*4882a593Smuzhiyun 		pr_info("config cancel id = %d", rve_user_ctx_id);
376*4882a593Smuzhiyun 
377*4882a593Smuzhiyun 	/* find internal_ctx to set cmd by user ctx (internal ctx id) */
378*4882a593Smuzhiyun 	ret = rve_job_cancel_by_user_ctx(rve_user_ctx_id);
379*4882a593Smuzhiyun 	if (ret < 0) {
380*4882a593Smuzhiyun 		pr_err("cancel ctx id[%d] failed!\n", rve_user_ctx_id);
381*4882a593Smuzhiyun 		return -EFAULT;
382*4882a593Smuzhiyun 	}
383*4882a593Smuzhiyun 
384*4882a593Smuzhiyun 	return ret;
385*4882a593Smuzhiyun }
386*4882a593Smuzhiyun 
rve_ioctl(struct file * file,uint32_t cmd,unsigned long arg)387*4882a593Smuzhiyun static long rve_ioctl(struct file *file, uint32_t cmd, unsigned long arg)
388*4882a593Smuzhiyun {
389*4882a593Smuzhiyun 	struct rve_drvdata_t *rve = rve_drvdata;
390*4882a593Smuzhiyun 
391*4882a593Smuzhiyun 	int ret = 0;
392*4882a593Smuzhiyun 	int i = 0;
393*4882a593Smuzhiyun 	struct rve_version_t driver_version;
394*4882a593Smuzhiyun 	struct rve_hw_versions_t hw_versions;
395*4882a593Smuzhiyun 	struct rve_session *session = file->private_data;
396*4882a593Smuzhiyun 
397*4882a593Smuzhiyun 	if (!rve) {
398*4882a593Smuzhiyun 		pr_err("rve_drvdata is null, rve is not init\n");
399*4882a593Smuzhiyun 		return -ENODEV;
400*4882a593Smuzhiyun 	}
401*4882a593Smuzhiyun 
402*4882a593Smuzhiyun 	//if (DEBUGGER_EN(NONUSE))
403*4882a593Smuzhiyun 	//	return 0;
404*4882a593Smuzhiyun 
405*4882a593Smuzhiyun 	switch (cmd) {
406*4882a593Smuzhiyun 	case RVE_IOC_GET_HW_VER:
407*4882a593Smuzhiyun 		/* RVE hardware version */
408*4882a593Smuzhiyun 		hw_versions.size = rve->num_of_scheduler > RVE_HW_SIZE ?
409*4882a593Smuzhiyun 			RVE_HW_SIZE : rve->num_of_scheduler;
410*4882a593Smuzhiyun 
411*4882a593Smuzhiyun 		for (i = 0; i < hw_versions.size; i++) {
412*4882a593Smuzhiyun 			memcpy(&hw_versions.version[i], &rve->scheduler[i]->version,
413*4882a593Smuzhiyun 				sizeof(rve->scheduler[i]->version));
414*4882a593Smuzhiyun 		}
415*4882a593Smuzhiyun 
416*4882a593Smuzhiyun 		if (copy_to_user((void *)arg, &hw_versions, sizeof(hw_versions)))
417*4882a593Smuzhiyun 			ret = -EFAULT;
418*4882a593Smuzhiyun 		else
419*4882a593Smuzhiyun 			ret = true;
420*4882a593Smuzhiyun 
421*4882a593Smuzhiyun 		break;
422*4882a593Smuzhiyun 
423*4882a593Smuzhiyun 	case RVE_IOC_GET_VER:
424*4882a593Smuzhiyun 		/* Driver version */
425*4882a593Smuzhiyun 		driver_version.major = DRIVER_MAJOR_VERSION;
426*4882a593Smuzhiyun 		driver_version.minor = DRIVER_MINOR_VERSION;
427*4882a593Smuzhiyun 		driver_version.revision = DRIVER_REVISION_VERSION;
428*4882a593Smuzhiyun 		driver_version.prod_num = 0;
429*4882a593Smuzhiyun 		strncpy((char *)driver_version.str, DRIVER_VERSION, sizeof(driver_version.str));
430*4882a593Smuzhiyun 
431*4882a593Smuzhiyun 		if (copy_to_user((void *)arg, &driver_version, sizeof(driver_version)))
432*4882a593Smuzhiyun 			ret = -EFAULT;
433*4882a593Smuzhiyun 		else
434*4882a593Smuzhiyun 			ret = true;
435*4882a593Smuzhiyun 
436*4882a593Smuzhiyun 		break;
437*4882a593Smuzhiyun 
438*4882a593Smuzhiyun 	case RVE_IOC_START_CONFIG:
439*4882a593Smuzhiyun 		ret = rve_ioctl_cmd_start(arg, session);
440*4882a593Smuzhiyun 
441*4882a593Smuzhiyun 		break;
442*4882a593Smuzhiyun 
443*4882a593Smuzhiyun 	case RVE_IOC_END_CONFIG:
444*4882a593Smuzhiyun 		ret = rve_ioctl_cmd_end(arg);
445*4882a593Smuzhiyun 
446*4882a593Smuzhiyun 		break;
447*4882a593Smuzhiyun 
448*4882a593Smuzhiyun 	case RVE_IOC_CMD_CONFIG:
449*4882a593Smuzhiyun 		ret = rve_ioctl_cmd_config(arg);
450*4882a593Smuzhiyun 
451*4882a593Smuzhiyun 		break;
452*4882a593Smuzhiyun 
453*4882a593Smuzhiyun 	case RVE_IOC_CANCEL_CONFIG:
454*4882a593Smuzhiyun 		ret = rve_ioctl_cmd_cancel(arg);
455*4882a593Smuzhiyun 
456*4882a593Smuzhiyun 		break;
457*4882a593Smuzhiyun 
458*4882a593Smuzhiyun 	default:
459*4882a593Smuzhiyun 		pr_err("unknown ioctl cmd!\n");
460*4882a593Smuzhiyun 		ret = -EINVAL;
461*4882a593Smuzhiyun 		break;
462*4882a593Smuzhiyun 	}
463*4882a593Smuzhiyun 
464*4882a593Smuzhiyun 	return ret;
465*4882a593Smuzhiyun }
466*4882a593Smuzhiyun 
467*4882a593Smuzhiyun #ifdef CONFIG_ROCKCHIP_RVE_DEBUGGER
rve_debugger_init(struct rve_debugger ** debugger_p)468*4882a593Smuzhiyun static int rve_debugger_init(struct rve_debugger **debugger_p)
469*4882a593Smuzhiyun {
470*4882a593Smuzhiyun 	struct rve_debugger *debugger;
471*4882a593Smuzhiyun 
472*4882a593Smuzhiyun 	*debugger_p = kzalloc(sizeof(struct rve_debugger), GFP_KERNEL);
473*4882a593Smuzhiyun 	if (*debugger_p == NULL) {
474*4882a593Smuzhiyun 		pr_err("can not alloc for rve debugger\n");
475*4882a593Smuzhiyun 		return -ENOMEM;
476*4882a593Smuzhiyun 	}
477*4882a593Smuzhiyun 
478*4882a593Smuzhiyun 	debugger = *debugger_p;
479*4882a593Smuzhiyun 
480*4882a593Smuzhiyun #ifdef CONFIG_ROCKCHIP_RVE_DEBUG_FS
481*4882a593Smuzhiyun 	mutex_init(&debugger->debugfs_lock);
482*4882a593Smuzhiyun 	INIT_LIST_HEAD(&debugger->debugfs_entry_list);
483*4882a593Smuzhiyun #endif
484*4882a593Smuzhiyun 
485*4882a593Smuzhiyun #ifdef CONFIG_ROCKCHIP_RVE_PROC_FS
486*4882a593Smuzhiyun 	mutex_init(&debugger->procfs_lock);
487*4882a593Smuzhiyun 	INIT_LIST_HEAD(&debugger->procfs_entry_list);
488*4882a593Smuzhiyun #endif
489*4882a593Smuzhiyun 
490*4882a593Smuzhiyun 	rve_debugfs_init();
491*4882a593Smuzhiyun 	rve_procfs_init();
492*4882a593Smuzhiyun 
493*4882a593Smuzhiyun 	return 0;
494*4882a593Smuzhiyun }
495*4882a593Smuzhiyun 
rve_debugger_remove(struct rve_debugger ** debugger_p)496*4882a593Smuzhiyun static int rve_debugger_remove(struct rve_debugger **debugger_p)
497*4882a593Smuzhiyun {
498*4882a593Smuzhiyun 	rve_debugfs_remove();
499*4882a593Smuzhiyun 	rve_procfs_remove();
500*4882a593Smuzhiyun 
501*4882a593Smuzhiyun 	kfree(*debugger_p);
502*4882a593Smuzhiyun 	*debugger_p = NULL;
503*4882a593Smuzhiyun 
504*4882a593Smuzhiyun 	return 0;
505*4882a593Smuzhiyun }
506*4882a593Smuzhiyun #endif
507*4882a593Smuzhiyun 
rve_open(struct inode * inode,struct file * file)508*4882a593Smuzhiyun static int rve_open(struct inode *inode, struct file *file)
509*4882a593Smuzhiyun {
510*4882a593Smuzhiyun 	struct rve_session *session = NULL;
511*4882a593Smuzhiyun 
512*4882a593Smuzhiyun 	session = rve_session_init();
513*4882a593Smuzhiyun 	if (!session)
514*4882a593Smuzhiyun 		return -ENOMEM;
515*4882a593Smuzhiyun 
516*4882a593Smuzhiyun 	file->private_data = (void *)session;
517*4882a593Smuzhiyun 
518*4882a593Smuzhiyun 	return nonseekable_open(inode, file);
519*4882a593Smuzhiyun }
520*4882a593Smuzhiyun 
rve_release(struct inode * inode,struct file * file)521*4882a593Smuzhiyun static int rve_release(struct inode *inode, struct file *file)
522*4882a593Smuzhiyun {
523*4882a593Smuzhiyun 	struct rve_session *session = file->private_data;
524*4882a593Smuzhiyun 
525*4882a593Smuzhiyun 	rve_session_deinit(session);
526*4882a593Smuzhiyun 
527*4882a593Smuzhiyun 	return 0;
528*4882a593Smuzhiyun }
529*4882a593Smuzhiyun 
rve_irq_handler(int irq,void * data)530*4882a593Smuzhiyun static irqreturn_t rve_irq_handler(int irq, void *data)
531*4882a593Smuzhiyun {
532*4882a593Smuzhiyun 	struct rve_scheduler_t *scheduler = data;
533*4882a593Smuzhiyun 	u32 error_flag;
534*4882a593Smuzhiyun 
535*4882a593Smuzhiyun 	error_flag = rve_read(RVE_SWREG6_IVE_WORK_STA, scheduler);
536*4882a593Smuzhiyun 
537*4882a593Smuzhiyun 	if (error_flag & 0x6) {
538*4882a593Smuzhiyun 		pr_err("irq thread work_status[%x]\n", error_flag);
539*4882a593Smuzhiyun 
540*4882a593Smuzhiyun 		if (error_flag & 0x2)
541*4882a593Smuzhiyun 			pr_err("irq: bus error");
542*4882a593Smuzhiyun 		else if (error_flag & 0x4)
543*4882a593Smuzhiyun 			pr_err("irq: timeout error");
544*4882a593Smuzhiyun 
545*4882a593Smuzhiyun 		scheduler->ops->soft_reset(scheduler);
546*4882a593Smuzhiyun 	}
547*4882a593Smuzhiyun 
548*4882a593Smuzhiyun 	/* clear INT */
549*4882a593Smuzhiyun 	rve_write(0x30000, RVE_SWREG1_IVE_IRQ, scheduler);
550*4882a593Smuzhiyun 
551*4882a593Smuzhiyun 	return IRQ_WAKE_THREAD;
552*4882a593Smuzhiyun }
553*4882a593Smuzhiyun 
rve_irq_thread(int irq,void * data)554*4882a593Smuzhiyun static irqreturn_t rve_irq_thread(int irq, void *data)
555*4882a593Smuzhiyun {
556*4882a593Smuzhiyun 	struct rve_scheduler_t *scheduler = data;
557*4882a593Smuzhiyun 	struct rve_job *job;
558*4882a593Smuzhiyun 	u32 error_flag;
559*4882a593Smuzhiyun 
560*4882a593Smuzhiyun 	job = scheduler->running_job;
561*4882a593Smuzhiyun 	scheduler->session.total_int_cnt++;
562*4882a593Smuzhiyun 
563*4882a593Smuzhiyun 	if (!job) {
564*4882a593Smuzhiyun 		pr_err("running job is invalid on irq thread\n");
565*4882a593Smuzhiyun 		return IRQ_HANDLED;
566*4882a593Smuzhiyun 	}
567*4882a593Smuzhiyun 
568*4882a593Smuzhiyun 	error_flag = rve_read(RVE_SWREG6_IVE_WORK_STA, scheduler);
569*4882a593Smuzhiyun 
570*4882a593Smuzhiyun 	if (DEBUGGER_EN(INT_FLAG)) {
571*4882a593Smuzhiyun 		pr_err("irq thread work_status[%x]\n", error_flag);
572*4882a593Smuzhiyun 		if (error_flag & 0x6) {
573*4882a593Smuzhiyun 			if (error_flag & 0x2)
574*4882a593Smuzhiyun 				pr_err("irq: bus error");
575*4882a593Smuzhiyun 			else if (error_flag & 0x4)
576*4882a593Smuzhiyun 				pr_err("irq: timeout error");
577*4882a593Smuzhiyun 		}
578*4882a593Smuzhiyun 	}
579*4882a593Smuzhiyun 
580*4882a593Smuzhiyun 	/* if llp mode*/
581*4882a593Smuzhiyun 	if ((error_flag & RVE_LLP_MODE) &&
582*4882a593Smuzhiyun 	    (!(error_flag & RVE_LLP_DONE))) {
583*4882a593Smuzhiyun 		if (DEBUGGER_EN(INT_FLAG))
584*4882a593Smuzhiyun 			pr_err("irq: llp mode need to skip rve_job_done");
585*4882a593Smuzhiyun 		goto skip_job_done;
586*4882a593Smuzhiyun 	}
587*4882a593Smuzhiyun 
588*4882a593Smuzhiyun 	rve_job_done(scheduler, 0);
589*4882a593Smuzhiyun 
590*4882a593Smuzhiyun skip_job_done:
591*4882a593Smuzhiyun 	return IRQ_HANDLED;
592*4882a593Smuzhiyun }
593*4882a593Smuzhiyun 
594*4882a593Smuzhiyun const struct file_operations rve_fops = {
595*4882a593Smuzhiyun 	.owner = THIS_MODULE,
596*4882a593Smuzhiyun 	.open = rve_open,
597*4882a593Smuzhiyun 	.release = rve_release,
598*4882a593Smuzhiyun 	.unlocked_ioctl = rve_ioctl,
599*4882a593Smuzhiyun #ifdef CONFIG_COMPAT
600*4882a593Smuzhiyun 	.compat_ioctl = rve_ioctl,
601*4882a593Smuzhiyun #endif
602*4882a593Smuzhiyun };
603*4882a593Smuzhiyun 
604*4882a593Smuzhiyun static struct miscdevice rve_dev = {
605*4882a593Smuzhiyun 	.minor = MISC_DYNAMIC_MINOR,
606*4882a593Smuzhiyun 	.name = "rve",
607*4882a593Smuzhiyun 	.fops = &rve_fops,
608*4882a593Smuzhiyun };
609*4882a593Smuzhiyun 
610*4882a593Smuzhiyun static const char *const rve_clks[] = {
611*4882a593Smuzhiyun 	"aclk_rve",
612*4882a593Smuzhiyun 	"hclk_rve",
613*4882a593Smuzhiyun };
614*4882a593Smuzhiyun 
615*4882a593Smuzhiyun static const struct rve_irqs_data_t rve_irqs[] = {
616*4882a593Smuzhiyun 	{"rve_irq", rve_irq_handler, rve_irq_thread}
617*4882a593Smuzhiyun };
618*4882a593Smuzhiyun 
619*4882a593Smuzhiyun static const struct rve_match_data_t rve_match_data = {
620*4882a593Smuzhiyun 	.clks = rve_clks,
621*4882a593Smuzhiyun 	.num_clks = ARRAY_SIZE(rve_clks),
622*4882a593Smuzhiyun 	.irqs = rve_irqs,
623*4882a593Smuzhiyun 	.num_irqs = ARRAY_SIZE(rve_irqs)
624*4882a593Smuzhiyun };
625*4882a593Smuzhiyun 
626*4882a593Smuzhiyun static const struct of_device_id rve_dt_ids[] = {
627*4882a593Smuzhiyun 	{
628*4882a593Smuzhiyun 	 .compatible = "rockchip,rve",
629*4882a593Smuzhiyun 	 .data = &rve_match_data,
630*4882a593Smuzhiyun 	},
631*4882a593Smuzhiyun 	{},
632*4882a593Smuzhiyun };
633*4882a593Smuzhiyun 
init_scheduler(struct rve_scheduler_t * scheduler,const char * name)634*4882a593Smuzhiyun static void init_scheduler(struct rve_scheduler_t *scheduler,
635*4882a593Smuzhiyun 			 const char *name)
636*4882a593Smuzhiyun {
637*4882a593Smuzhiyun 	spin_lock_init(&scheduler->irq_lock);
638*4882a593Smuzhiyun 	INIT_LIST_HEAD(&scheduler->todo_list);
639*4882a593Smuzhiyun 	init_waitqueue_head(&scheduler->job_done_wq);
640*4882a593Smuzhiyun 
641*4882a593Smuzhiyun 	if (!strcmp(name, "rve")) {
642*4882a593Smuzhiyun 		scheduler->ops = &rve_ops;
643*4882a593Smuzhiyun 		scheduler->core = RVE_SCHEDULER_CORE0;
644*4882a593Smuzhiyun 	}
645*4882a593Smuzhiyun }
646*4882a593Smuzhiyun 
rve_drv_probe(struct platform_device * pdev)647*4882a593Smuzhiyun static int rve_drv_probe(struct platform_device *pdev)
648*4882a593Smuzhiyun {
649*4882a593Smuzhiyun 	struct rve_drvdata_t *data = rve_drvdata;
650*4882a593Smuzhiyun 	struct resource *res;
651*4882a593Smuzhiyun 	int ret = 0;
652*4882a593Smuzhiyun 	const struct of_device_id *match = NULL;
653*4882a593Smuzhiyun 	struct device *dev = &pdev->dev;
654*4882a593Smuzhiyun 	const struct rve_match_data_t *match_data;
655*4882a593Smuzhiyun 	int i = 0, irq;
656*4882a593Smuzhiyun 	struct rve_scheduler_t *scheduler = NULL;
657*4882a593Smuzhiyun 
658*4882a593Smuzhiyun 	if (!pdev->dev.of_node)
659*4882a593Smuzhiyun 		return -EINVAL;
660*4882a593Smuzhiyun 
661*4882a593Smuzhiyun 	if (!strcmp(dev_driver_string(dev), "rve"))
662*4882a593Smuzhiyun 		match = of_match_device(rve_dt_ids, dev);
663*4882a593Smuzhiyun 
664*4882a593Smuzhiyun 	if (!match) {
665*4882a593Smuzhiyun 		dev_err(dev, "%s missing DT entry!\n", dev_driver_string(dev));
666*4882a593Smuzhiyun 		return -EINVAL;
667*4882a593Smuzhiyun 	}
668*4882a593Smuzhiyun 
669*4882a593Smuzhiyun 	scheduler =
670*4882a593Smuzhiyun 		devm_kzalloc(&pdev->dev, sizeof(struct rve_scheduler_t),
671*4882a593Smuzhiyun 			GFP_KERNEL);
672*4882a593Smuzhiyun 	if (scheduler == NULL) {
673*4882a593Smuzhiyun 		pr_err("failed to allocate scheduler. dev name = %s\n",
674*4882a593Smuzhiyun 			dev_driver_string(dev));
675*4882a593Smuzhiyun 		return -ENOMEM;
676*4882a593Smuzhiyun 	}
677*4882a593Smuzhiyun 
678*4882a593Smuzhiyun 	init_scheduler(scheduler, dev_driver_string(dev));
679*4882a593Smuzhiyun 
680*4882a593Smuzhiyun 	scheduler->dev = &pdev->dev;
681*4882a593Smuzhiyun 
682*4882a593Smuzhiyun 	/* map the registers */
683*4882a593Smuzhiyun 	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
684*4882a593Smuzhiyun 	if (!res) {
685*4882a593Smuzhiyun 		pr_err("get memory resource failed.\n");
686*4882a593Smuzhiyun 		return -ENXIO;
687*4882a593Smuzhiyun 	}
688*4882a593Smuzhiyun 
689*4882a593Smuzhiyun 	scheduler->rve_base =
690*4882a593Smuzhiyun 		devm_ioremap(&pdev->dev, res->start, resource_size(res));
691*4882a593Smuzhiyun 	if (!scheduler->rve_base) {
692*4882a593Smuzhiyun 		pr_err("ioremap failed\n");
693*4882a593Smuzhiyun 		ret = -ENOENT;
694*4882a593Smuzhiyun 		return ret;
695*4882a593Smuzhiyun 	}
696*4882a593Smuzhiyun 
697*4882a593Smuzhiyun 	/* get the IRQ */
698*4882a593Smuzhiyun 	match_data = match->data;
699*4882a593Smuzhiyun 
700*4882a593Smuzhiyun 	/* there are irq names in dts */
701*4882a593Smuzhiyun 	irq = platform_get_irq(pdev, 0);
702*4882a593Smuzhiyun 	if (irq < 0) {
703*4882a593Smuzhiyun 		dev_err(dev, "no irq %s in dts\n", match_data->irqs[0].name);
704*4882a593Smuzhiyun 		return irq;
705*4882a593Smuzhiyun 	}
706*4882a593Smuzhiyun 
707*4882a593Smuzhiyun 	scheduler->irq = irq;
708*4882a593Smuzhiyun 
709*4882a593Smuzhiyun 	pr_info("%s, irq = %d, match scheduler\n",
710*4882a593Smuzhiyun 			match_data->irqs[0].name, irq);
711*4882a593Smuzhiyun 
712*4882a593Smuzhiyun 	ret = devm_request_threaded_irq(dev, irq,
713*4882a593Smuzhiyun 			match_data->irqs[0].irq_hdl,
714*4882a593Smuzhiyun 			match_data->irqs[0].irq_thread, IRQF_SHARED,
715*4882a593Smuzhiyun 			dev_driver_string(dev), scheduler);
716*4882a593Smuzhiyun 	if (ret < 0) {
717*4882a593Smuzhiyun 		pr_err("request irq name: %s failed: %d\n",
718*4882a593Smuzhiyun 				match_data->irqs[0].name, ret);
719*4882a593Smuzhiyun 		return ret;
720*4882a593Smuzhiyun 	}
721*4882a593Smuzhiyun 
722*4882a593Smuzhiyun #ifndef RVE_PD_AWAYS_ON
723*4882a593Smuzhiyun 	for (i = 0; i < match_data->num_clks; i++) {
724*4882a593Smuzhiyun 		struct clk *clk = devm_clk_get(dev, match_data->clks[i]);
725*4882a593Smuzhiyun 
726*4882a593Smuzhiyun 		if (IS_ERR(clk))
727*4882a593Smuzhiyun 			pr_err("failed to get %s\n", match_data->clks[i]);
728*4882a593Smuzhiyun 
729*4882a593Smuzhiyun 		scheduler->clks[i] = clk;
730*4882a593Smuzhiyun 	}
731*4882a593Smuzhiyun 	scheduler->num_clks = match_data->num_clks;
732*4882a593Smuzhiyun #endif
733*4882a593Smuzhiyun 
734*4882a593Smuzhiyun 	platform_set_drvdata(pdev, scheduler);
735*4882a593Smuzhiyun 
736*4882a593Smuzhiyun 	device_init_wakeup(dev, true);
737*4882a593Smuzhiyun 
738*4882a593Smuzhiyun 	/* PM init */
739*4882a593Smuzhiyun #ifndef RVE_PD_AWAYS_ON
740*4882a593Smuzhiyun 	pm_runtime_enable(&pdev->dev);
741*4882a593Smuzhiyun 
742*4882a593Smuzhiyun 	ret = pm_runtime_get_sync(scheduler->dev);
743*4882a593Smuzhiyun 	if (ret < 0) {
744*4882a593Smuzhiyun 		pr_err("failed to get pm runtime, ret = %d\n",
745*4882a593Smuzhiyun 			 ret);
746*4882a593Smuzhiyun 		goto failed;
747*4882a593Smuzhiyun 	}
748*4882a593Smuzhiyun 
749*4882a593Smuzhiyun 	for (i = 0; i < scheduler->num_clks; i++) {
750*4882a593Smuzhiyun 		if (!IS_ERR(scheduler->clks[i])) {
751*4882a593Smuzhiyun 			ret = clk_prepare_enable(scheduler->clks[i]);
752*4882a593Smuzhiyun 			if (ret < 0) {
753*4882a593Smuzhiyun 				pr_err("failed to enable clk\n");
754*4882a593Smuzhiyun 				goto failed;
755*4882a593Smuzhiyun 			}
756*4882a593Smuzhiyun 		}
757*4882a593Smuzhiyun 	}
758*4882a593Smuzhiyun #endif //RVE_PD_AWAYS_ON
759*4882a593Smuzhiyun 
760*4882a593Smuzhiyun 	scheduler->ops->get_version(scheduler);
761*4882a593Smuzhiyun 	pr_info("Driver loaded successfully rve[%d] ver:%s\n", i,
762*4882a593Smuzhiyun 		scheduler->version.str);
763*4882a593Smuzhiyun 
764*4882a593Smuzhiyun 	data->scheduler[data->num_of_scheduler] = scheduler;
765*4882a593Smuzhiyun 
766*4882a593Smuzhiyun 	data->num_of_scheduler++;
767*4882a593Smuzhiyun 
768*4882a593Smuzhiyun #ifndef RVE_PD_AWAYS_ON
769*4882a593Smuzhiyun 	for (i = scheduler->num_clks - 1; i >= 0; i--)
770*4882a593Smuzhiyun 		if (!IS_ERR(scheduler->clks[i]))
771*4882a593Smuzhiyun 			clk_disable_unprepare(scheduler->clks[i]);
772*4882a593Smuzhiyun 
773*4882a593Smuzhiyun 	pm_runtime_put_sync(&pdev->dev);
774*4882a593Smuzhiyun #endif //RVE_PD_AWAYS_ON
775*4882a593Smuzhiyun 
776*4882a593Smuzhiyun 	pr_info("probe successfully\n");
777*4882a593Smuzhiyun 
778*4882a593Smuzhiyun 	return 0;
779*4882a593Smuzhiyun 
780*4882a593Smuzhiyun #ifndef RVE_PD_AWAYS_ON
781*4882a593Smuzhiyun failed:
782*4882a593Smuzhiyun 	device_init_wakeup(dev, false);
783*4882a593Smuzhiyun 	pm_runtime_disable(dev);
784*4882a593Smuzhiyun 
785*4882a593Smuzhiyun 	return ret;
786*4882a593Smuzhiyun #endif //RVE_PD_AWAYS_ON
787*4882a593Smuzhiyun }
788*4882a593Smuzhiyun 
rve_drv_remove(struct platform_device * pdev)789*4882a593Smuzhiyun static int rve_drv_remove(struct platform_device *pdev)
790*4882a593Smuzhiyun {
791*4882a593Smuzhiyun 	device_init_wakeup(&pdev->dev, false);
792*4882a593Smuzhiyun #ifndef RVE_PD_AWAYS_ON
793*4882a593Smuzhiyun 	pm_runtime_disable(&pdev->dev);
794*4882a593Smuzhiyun #endif //RVE_PD_AWAYS_ON
795*4882a593Smuzhiyun 
796*4882a593Smuzhiyun 	return 0;
797*4882a593Smuzhiyun }
798*4882a593Smuzhiyun 
799*4882a593Smuzhiyun static struct platform_driver rve_driver = {
800*4882a593Smuzhiyun 	.probe = rve_drv_probe,
801*4882a593Smuzhiyun 	.remove = rve_drv_remove,
802*4882a593Smuzhiyun 	.driver = {
803*4882a593Smuzhiyun 		 .name = "rve",
804*4882a593Smuzhiyun 		 .of_match_table = of_match_ptr(rve_dt_ids),
805*4882a593Smuzhiyun 		 },
806*4882a593Smuzhiyun };
807*4882a593Smuzhiyun 
rve_init(void)808*4882a593Smuzhiyun static int __init rve_init(void)
809*4882a593Smuzhiyun {
810*4882a593Smuzhiyun 	int ret;
811*4882a593Smuzhiyun 
812*4882a593Smuzhiyun 	rve_drvdata = kzalloc(sizeof(struct rve_drvdata_t), GFP_KERNEL);
813*4882a593Smuzhiyun 	if (rve_drvdata == NULL) {
814*4882a593Smuzhiyun 		pr_err("failed to allocate driver data.\n");
815*4882a593Smuzhiyun 		return -ENOMEM;
816*4882a593Smuzhiyun 	}
817*4882a593Smuzhiyun 
818*4882a593Smuzhiyun 	mutex_init(&rve_drvdata->lock);
819*4882a593Smuzhiyun 
820*4882a593Smuzhiyun 	wake_lock_init(&rve_drvdata->wake_lock, WAKE_LOCK_SUSPEND, "rve");
821*4882a593Smuzhiyun 
822*4882a593Smuzhiyun 	ret = platform_driver_register(&rve_driver);
823*4882a593Smuzhiyun 	if (ret != 0) {
824*4882a593Smuzhiyun 		pr_err("Platform device rve register failed (%d).\n", ret);
825*4882a593Smuzhiyun 		return ret;
826*4882a593Smuzhiyun 	}
827*4882a593Smuzhiyun 
828*4882a593Smuzhiyun #ifdef CONFIG_SYNC_FILE
829*4882a593Smuzhiyun 	rve_drvdata->fence_ctx = rve_fence_context_alloc();
830*4882a593Smuzhiyun 	if (IS_ERR(rve_drvdata->fence_ctx)) {
831*4882a593Smuzhiyun 		pr_err("failed to allocate fence context for RVE\n");
832*4882a593Smuzhiyun 		ret = PTR_ERR(rve_drvdata->fence_ctx);
833*4882a593Smuzhiyun 		return ret;
834*4882a593Smuzhiyun 	}
835*4882a593Smuzhiyun #endif
836*4882a593Smuzhiyun 
837*4882a593Smuzhiyun 	ret = misc_register(&rve_dev);
838*4882a593Smuzhiyun 	if (ret) {
839*4882a593Smuzhiyun 		pr_err("cannot register miscdev (%d)\n", ret);
840*4882a593Smuzhiyun 		return ret;
841*4882a593Smuzhiyun 	}
842*4882a593Smuzhiyun 
843*4882a593Smuzhiyun 	rve_ctx_manager_init(&rve_drvdata->pend_ctx_manager);
844*4882a593Smuzhiyun 
845*4882a593Smuzhiyun 	rve_session_manager_init(&rve_drvdata->session_manager);
846*4882a593Smuzhiyun 
847*4882a593Smuzhiyun 	rve_init_timer();
848*4882a593Smuzhiyun 
849*4882a593Smuzhiyun #ifdef CONFIG_ROCKCHIP_RVE_DEBUGGER
850*4882a593Smuzhiyun 	rve_debugger_init(&rve_drvdata->debugger);
851*4882a593Smuzhiyun #endif
852*4882a593Smuzhiyun 
853*4882a593Smuzhiyun 	pr_info("Module initialized. v%s\n", DRIVER_VERSION);
854*4882a593Smuzhiyun 
855*4882a593Smuzhiyun 	return 0;
856*4882a593Smuzhiyun }
857*4882a593Smuzhiyun 
rve_exit(void)858*4882a593Smuzhiyun static void __exit rve_exit(void)
859*4882a593Smuzhiyun {
860*4882a593Smuzhiyun #ifdef CONFIG_ROCKCHIP_RVE_DEBUGGER
861*4882a593Smuzhiyun 	rve_debugger_remove(&rve_drvdata->debugger);
862*4882a593Smuzhiyun #endif
863*4882a593Smuzhiyun 
864*4882a593Smuzhiyun 	rve_ctx_manager_remove(&rve_drvdata->pend_ctx_manager);
865*4882a593Smuzhiyun 
866*4882a593Smuzhiyun 	rve_session_manager_remove(&rve_drvdata->session_manager);
867*4882a593Smuzhiyun 
868*4882a593Smuzhiyun 	wake_lock_destroy(&rve_drvdata->wake_lock);
869*4882a593Smuzhiyun 
870*4882a593Smuzhiyun #ifdef CONFIG_SYNC_FILE
871*4882a593Smuzhiyun 	rve_fence_context_free(rve_drvdata->fence_ctx);
872*4882a593Smuzhiyun #endif
873*4882a593Smuzhiyun 
874*4882a593Smuzhiyun 	rve_cancel_timer();
875*4882a593Smuzhiyun 
876*4882a593Smuzhiyun 	platform_driver_unregister(&rve_driver);
877*4882a593Smuzhiyun 
878*4882a593Smuzhiyun 	misc_deregister(&rve_dev);
879*4882a593Smuzhiyun 
880*4882a593Smuzhiyun 	kfree(rve_drvdata);
881*4882a593Smuzhiyun }
882*4882a593Smuzhiyun 
883*4882a593Smuzhiyun #if (LINUX_VERSION_CODE >= KERNEL_VERSION(4, 4, 0))
884*4882a593Smuzhiyun #ifdef CONFIG_ROCKCHIP_THUNDER_BOOT
885*4882a593Smuzhiyun module_init(rve_init);
886*4882a593Smuzhiyun #else
887*4882a593Smuzhiyun late_initcall(rve_init);
888*4882a593Smuzhiyun #endif
889*4882a593Smuzhiyun #else
890*4882a593Smuzhiyun fs_initcall(rve_init);
891*4882a593Smuzhiyun #endif
892*4882a593Smuzhiyun module_exit(rve_exit);
893*4882a593Smuzhiyun 
894*4882a593Smuzhiyun /* Module information */
895*4882a593Smuzhiyun MODULE_AUTHOR("putin.li@rock-chips.com");
896*4882a593Smuzhiyun MODULE_DESCRIPTION("Driver for rve device");
897*4882a593Smuzhiyun MODULE_LICENSE("GPL");
898