diff options
-rw-r--r-- | drivers/edgetpu/abrolhos-platform.h | 4 | ||||
-rw-r--r-- | drivers/edgetpu/abrolhos-pm.c | 6 |
2 files changed, 7 insertions, 3 deletions
diff --git a/drivers/edgetpu/abrolhos-platform.h b/drivers/edgetpu/abrolhos-platform.h index 73e2221..068a23f 100644 --- a/drivers/edgetpu/abrolhos-platform.h +++ b/drivers/edgetpu/abrolhos-platform.h @@ -52,7 +52,9 @@ struct abrolhos_platform_dev { struct edgetpu_coherent_mem log_mem; struct edgetpu_coherent_mem trace_mem; struct abrolhos_sscd_info sscd_info; - struct gs101_bcl_dev *bcl_dev; +#if IS_ENABLED(CONFIG_GOOGLE_BCL) + struct bcl_device *bcl_dev; +#endif /* Protects TZ Mailbox client pointer */ struct mutex tz_mailbox_lock; /* TZ mailbox client */ diff --git a/drivers/edgetpu/abrolhos-pm.c b/drivers/edgetpu/abrolhos-pm.c index 9a3db2f..651a831 100644 --- a/drivers/edgetpu/abrolhos-pm.c +++ b/drivers/edgetpu/abrolhos-pm.c @@ -492,10 +492,12 @@ static int abrolhos_power_up(struct edgetpu_pm *etpm) if (ret) { abrolhos_power_down(etpm); } else { +#if IS_ENABLED(CONFIG_GOOGLE_BCL) if (!abpdev->bcl_dev) - abpdev->bcl_dev = gs101_retrieve_bcl_handle(); + abpdev->bcl_dev = google_retrieve_bcl_handle(); if (abpdev->bcl_dev) - gs101_init_tpu_ratio(abpdev->bcl_dev); + google_init_tpu_ratio(abpdev->bcl_dev); +#endif } return ret; |