[NDTensors] Vendor TypeParameterAccessors.jl (#1676)

Author: mtfishman

NDTensors/Project.toml

@@ -1,7 +1,7 @@
 name = "NDTensors"
 uuid = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
 authors = ["Matthew Fishman <[email protected]>"]
-version = "0.4.13"
+version = "0.4.14"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
@@ -28,7 +28,6 @@ Strided = "5e0ebb24-38b0-5f93-81fe-25c709ecae67"
 StridedViews = "4db3bf67-4bd7-4b4e-b153-31dc3fb37143"
 TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 TupleTools = "9d95972d-f1c8-5527-a6e0-b4b365fa01f6"
-TypeParameterAccessors = "7e5a90cf-f82e-492e-a09b-e3e26432c138"
 VectorInterface = "409d34a3-91d5-4945-b6ec-7529ddf182d8"
 
 [weakdeps]
@@ -88,7 +87,6 @@ StridedViews = "0.2.2, 0.3, 0.4"
 TBLIS = "0.2"
 TimerOutputs = "0.5.5"
 TupleTools = "1.2.0"
-TypeParameterAccessors = "0.3"
 VectorInterface = "0.4.2, 0.5"
 cuTENSOR = "2"
 julia = "1.10"

NDTensors/ext/NDTensorsAMDGPUExt/NDTensorsAMDGPUExt.jl

@@ -1,5 +1,14 @@
 module NDTensorsAMDGPUExt
 
+module Vendored
+    include(
+        joinpath(
+            "..", "..", "src", "vendored", "TypeParameterAccessors", "ext",
+            "TypeParameterAccessorsAMDGPUExt.jl"
+        )
+    )
+end
+
 include("append.jl")
 include("copyto.jl")
 include("set_types.jl")

NDTensors/ext/NDTensorsAMDGPUExt/adapt.jl

@@ -1,7 +1,7 @@
 using NDTensors: NDTensors, EmptyStorage, adapt_storagetype, emptytype
 using NDTensors.AMDGPUExtensions: AMDGPUExtensions, ROCArrayAdaptor
 using NDTensors.GPUArraysCoreExtensions: storagemode
-using NDTensors.TypeParameterAccessors:
+using NDTensors.Vendored.TypeParameterAccessors:
     default_type_parameters, set_type_parameters, type_parameters
 using Adapt: Adapt, adapt
 using AMDGPU: AMDGPU, ROCArray, ROCVector

NDTensors/ext/NDTensorsAMDGPUExt/linearalgebra.jl

@@ -1,7 +1,7 @@
 using NDTensors.AMDGPUExtensions: roc
 using NDTensors.Expose: Expose, Exposed, expose, ql, ql_positive
 using NDTensors.GPUArraysCoreExtensions: cpu
-using NDTensors.TypeParameterAccessors: unwrap_array_type
+using NDTensors.Vendored.TypeParameterAccessors: unwrap_array_type
 using LinearAlgebra: svd
 using Adapt: adapt
 using AMDGPU: ROCMatrix

NDTensors/ext/NDTensorsAMDGPUExt/set_types.jl

@@ -1,5 +1,5 @@
 using AMDGPU: ROCArray
 using NDTensors.GPUArraysCoreExtensions: storagemode
-using NDTensors.TypeParameterAccessors: TypeParameterAccessors, Position
+using NDTensors.Vendored.TypeParameterAccessors: TypeParameterAccessors, Position
 
 TypeParameterAccessors.position(::Type{<:ROCArray}, ::typeof(storagemode)) = Position(3)

NDTensors/ext/NDTensorsCUDAExt/NDTensorsCUDAExt.jl

@@ -1,4 +1,14 @@
 module NDTensorsCUDAExt
+
+module Vendored
+    include(
+        joinpath(
+            "..", "..", "src", "vendored", "TypeParameterAccessors", "ext",
+            "TypeParameterAccessorsCUDAExt.jl"
+        )
+    )
+end
+
 include("append.jl")
 include("default_kwargs.jl")
 include("copyto.jl")
@@ -9,4 +19,5 @@ include("indexing.jl")
 include("linearalgebra.jl")
 include("mul.jl")
 include("permutedims.jl")
+
 end

NDTensors/ext/NDTensorsCUDAExt/adapt.jl

@@ -4,7 +4,7 @@ using Functors: fmap
 using NDTensors: NDTensors, EmptyStorage, adapt_storagetype, emptytype
 using NDTensors.CUDAExtensions: CUDAExtensions, CuArrayAdaptor
 using NDTensors.GPUArraysCoreExtensions: storagemode
-using NDTensors.TypeParameterAccessors:
+using NDTensors.Vendored.TypeParameterAccessors:
     default_type_parameters, set_type_parameters, type_parameters
 
 function CUDAExtensions.cu(xs; storagemode = default_type_parameters(CuArray, storagemode))

NDTensors/ext/NDTensorsCUDAExt/linearalgebra.jl

@@ -4,7 +4,7 @@ using LinearAlgebra: Adjoint, svd
 using NDTensors: NDTensors
 using NDTensors.Expose: Expose, expose, ql, ql_positive
 using NDTensors.GPUArraysCoreExtensions: cpu
-using NDTensors.TypeParameterAccessors: unwrap_array_type
+using NDTensors.Vendored.TypeParameterAccessors: unwrap_array_type
 function NDTensors.svd_catch_error(A::CuMatrix; alg::String = "jacobi_algorithm")
     if alg == "jacobi_algorithm"
         alg = CUDA.CUSOLVER.JacobiAlgorithm()

NDTensors/ext/NDTensorsCUDAExt/set_types.jl

@@ -1,6 +1,6 @@
 using CUDA: CuArray
 using NDTensors.GPUArraysCoreExtensions: storagemode
-using NDTensors.TypeParameterAccessors: TypeParameterAccessors, Position
+using NDTensors.Vendored.TypeParameterAccessors: TypeParameterAccessors, Position
 
 function TypeParameterAccessors.position(::Type{<:CuArray}, ::typeof(storagemode))
     return Position(3)

NDTensors/ext/NDTensorsGPUArraysCoreExt/contract.jl

@@ -2,7 +2,7 @@ using Adapt: adapt
 using GPUArraysCore: AbstractGPUArray
 using NDTensors: NDTensors, DenseTensor, DiagTensor, contract!, dense, inds, Tensor
 using NDTensors.Expose: Exposed, expose, unexpose
-using NDTensors.TypeParameterAccessors: parenttype, set_ndims
+using NDTensors.Vendored.TypeParameterAccessors: parenttype, set_ndims
 
 function NDTensors.contract!(
         output_tensor::Exposed{<:AbstractGPUArray, <:DenseTensor},