[krasovsky]

Так а в чем проблем?
Проверяй каждый раз при загрузке страницы ее ширину

window.onload = function(){
if (window.innerWidth >= 320) {
div.style.width = window.innerHeight + "px";
}
}