1 files changed, 4 insertions, 1 deletions

diff --git a/conf/machine/arria10.conf b/conf/machine/arria10.conf
index 40b258f..02b73df 100644
--- a/conf/machine/arria10.conf
+++ b/conf/machine/arria10.conf
@@ -21,7 +21,10 @@ KERNEL_DEVICETREE ?= "\
                         socfpga_arria10_socdk_qspi.dtb \
                         socfpga_arria10_socdk_nand.dtb \
                         "
-EXTRA_IMAGECMD_jffs2 = "--squash -s 256 -e 64KiB"
+
+JFFS2_ERASEBLOCK ?= "${@bb.utils.contains("A10_IMAGE_TYPE", "NAND", "-e 0x20000", "", d)} \
+                     ${@bb.utils.contains("A10_IMAGE_TYPE", "QSPI", "-e 0x10000", "", d)}"
+EXTRA_IMAGECMD_jffs2 ?= "--squash -s 256 ${JFFS2_ERASEBLOCK}"
 
 SERIAL_CONSOLES ?= "115200;ttyS0"