https://www.w3.org/TR/wai-aria-practices/examples/

This has been an excellent resource on something I've been working on, but I'm not sure what to do with aria attributes on a three column web app UI that collapses to a single column on small screens. What exactly is the correct way to go about it here? Was thinking originally of using role="grid" using the layout grid example for guidance and everything that applies with that but that doesn't really apply anymore when it's one column and now has navigation buttons.