diff --git a/include/linux/leds-qti-flash.h b/include/linux/leds-qti-flash.h
index 337ba0369f91..0f5b9dc9e0ce 100644
--- a/include/linux/leds-qti-flash.h
+++ b/include/linux/leds-qti-flash.h
@@ -1,6 +1,6 @@
 /* SPDX-License-Identifier: GPL-2.0-only */
 /*
- * Copyright (c) 2020, The Linux Foundation. All rights reserved.
+ * Copyright (c) 2020-2021, The Linux Foundation. All rights reserved.
  */
 
 #ifndef __LEDS_QTI_FLASH_H
@@ -35,7 +35,18 @@ static inline int qti_flash_led_prepare(struct led_trigger *trig,
 }
 
 static inline int qti_flash_led_set_param(struct led_trigger *trig,
-			struct flash_led_param param);
+			struct flash_led_param param)
+{
+	return -EINVAL;
+}
+#endif
+
+#if IS_ENABLED(CONFIG_BACKLIGHT_QCOM_SPMI_WLED)
+int wled_flash_led_prepare(struct led_trigger *trig, int options,
+					int *max_current);
+#else
+static inline int wled_flash_led_prepare(struct led_trigger *trig, int options,
+					int *max_current)
 {
 	return -EINVAL;
 }