fix return size

Author: perazz

src/stdlib_linalg_norms.fypp

@@ -488,6 +488,7 @@ ${loop_variables_end(rank-1," "*12)}$
         call parse_norm_type(order,norm_request,err_)
         call lange_task_request(norm_request,lange_task,err_)
         if (err_%error()) then 
+            allocate(nrm${emptyranksuffix(rank-2)}$)
             call linalg_error_handling(err_,err)
             return
         endif             
@@ -496,7 +497,7 @@ ${loop_variables_end(rank-1," "*12)}$
         s = shape(a,kind=ilp)
 
         ! Check if input column data is contiguous
-        contiguous_data = dims(1)==1
+        contiguous_data = all(dims==[1,2])
 
         ! Matrix norm size
         m = s(dims(1))
@@ -505,20 +506,17 @@ ${loop_variables_end(rank-1," "*12)}$
         ! Get packed data with norm dimensions as 1:2
         if (contiguous_data) then 
 
-            ! Collapse everything before the 1st dimension as apack's dim #1
-            ! Set size==1 for all unused trailing dimensions            
-            spack  = [product(s(1:dims(2)-1)),s(dims(2):),(1_ilp,j=1,dims(2)-2)]
-            
             ! Reshape without moving data
-            apack${shape_from_array_data('spack',rank)}$ => a
+            spack =  s
+            apack => a
 
         else
 
             ! Dimension permutations to map dims(1),dims(2) => 1:2
             perm = [dims,pack(dim_range, dim_range/=dims(1) .and. dim_range/=dims(2))]            
             iperm(perm) = dim_range            
             spack = s(perm)            
-            apack = reshape(a, shape=spack, order=iperm)                             
+            allocate(apack,source=reshape(a, shape=spack, order=iperm))
 
         endif