use space_isequal for Mul axis checks

Author: lkdvos

Project.toml

@@ -4,6 +4,7 @@ authors = ["ITensor developers <[email protected]> and contributors"]
 version = "0.4.19"
 
 [deps]
+ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
 BlockSparseArrays = "2c9a651f-6452-4ace-a6ac-809f4280fbb4"
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
@@ -23,6 +24,7 @@ TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
 GradedArraysTensorAlgebraExt = "TensorAlgebra"
 
 [compat]
+ArrayLayouts = "1.11.2"
 BlockArrays = "1.6"
 BlockSparseArrays = "0.8, 0.9.3"
 Compat = "4.16"

src/gradedarray.jl

@@ -9,6 +9,7 @@ using BlockSparseArrays:
   sparsemortar
 using LinearAlgebra: Adjoint
 using TypeParameterAccessors: similartype, unwrap_array_type
+using ArrayLayouts: ArrayLayouts
 
 const GradedArray{T,N,A<:AbstractArray{T,N},Blocks<:AbstractArray{A,N},Axes<:Tuple{AbstractGradedUnitRange{<:Integer},Vararg{AbstractGradedUnitRange{<:Integer}}}} = BlockSparseArray{
   T,N,A,Blocks,Axes
@@ -236,3 +237,11 @@ function Base.showarg(io::IO, a::GradedArray, toplevel::Bool)
   print(io, concretetype_to_string_truncated(typeof(a); param_truncation_length=40))
   return nothing
 end
+
+const AnyGradedMatrix{T} = Union{GradedMatrix{T},Adjoint{T,<:GradedMatrix{T}}}
+
+function ArrayLayouts._check_mul_axes(A::AnyGradedMatrix, B::AnyGradedMatrix)
+  axA = axes(A, 2)
+  axB = axes(B, 1)
+  return space_isequal(dual(axA), axB) || ArrayLayouts.throw_mul_axes_err(axA, axB)
+end