add parameter to set driver memory