dma: fix assertion check, which should only check 2D and 3D cases
dma: fix assertion check, which should only check 2D and 3D cases

StanfordLegion/legion,StanfordLegion/legion,StanfordLegion/legion,StanfordLegion/legion,StanfordLegion/legion,StanfordLegion/legion,StanfordLegion/legion,StanfordLegion/legion