Measurement of prompt photon production in $\sqrts_\mathrmNN = 8.16$ TeV $p$+Pb collisions with ATLAS