[krasovsky]

Можно попробовать что то типо этого

if (window.innerWidth <= 320){
     div.style.width = window.innerHeight + "px";        
}
else {
    div.style.width = '100%'; //если ширина не <= 320 установить ее в 100% или подставь свое значение
}