Add custom kernel image, dtb and modules support

Add option to pick custom kernel prebuilts from
prebuilt-kernel/android-$(CUSTOM) directory. This
helps in testing custom kernels. Default kernel
is still based on android-5.4.

Build with "make TARGET_KERNEL_USE=$(CUSTOM)".
eg. Run "make TARGET_KERNEL_USE=mainline' to pick kernel
    prebuilts from prebuilt-kernel/android-mainline
    directory.

Signed-off-by: Amit Pundir <pundiramit@gmail.com>
diff --git a/build/tasks/kernel.mk b/build/tasks/kernel.mk
index 99ee38e..0473f90 100644
--- a/build/tasks/kernel.mk
+++ b/build/tasks/kernel.mk
@@ -1,7 +1,12 @@
 ifneq ($(filter aosp_beryllium, $(TARGET_DEVICE)),)
 
+ifndef TARGET_KERNEL_USE
 IMAGE_GZ := device/linaro/dragonboard-kernel/android-5.4/Image.gz
 DTB := device/xiaomi/beryllium/prebuilt-kernel/android-5.4/sdm845-beryllium.dtb
+else
+IMAGE_GZ := device/xiaomi/beryllium/prebuilt-kernel/android-$(TARGET_KERNEL_USE)/Image.gz
+DTB := device/xiaomi/beryllium/prebuilt-kernel/android-$(TARGET_KERNEL_USE)/sdm845-beryllium.dtb
+endif
 
 $(PRODUCT_OUT)/kernel: $(IMAGE_GZ) $(DTB)
 	cat $(IMAGE_GZ) $(DTB) > $@