Add registryt to `TypeParameterAcessors

ITensor · Jun 21, 2024 · f23d305 · f23d305
1 parent c4dad63
commit f23d305
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/NDTensors/ext/NDTEnsorsJLArraysExt/NDTensorsJLArraysExt.jl b/NDTensors/ext/NDTEnsorsJLArraysExt/NDTensorsJLArraysExt.jl
@@ -1,3 +1,3 @@
 module NDTensorsJLArraysExt
-
+include("set_types.jl")
 end
diff --git a/NDTensors/ext/NDTEnsorsJLArraysExt/set_types.jl b/NDTensors/ext/NDTEnsorsJLArraysExt/set_types.jl
@@ -0,0 +1,10 @@
+# TypeParameterAccessors definitions
+using NDTensors.TypeParameterAccessors: TypeParameterAccessors, Position
+using NDTensors.GPUArraysCoreExtensions: storagemode
+using JLArrays: JLArrays, JLArray
+
+function TypeParameterAccessors.default_type_parameters(::Type{<:JLArray})
+  return (Float64, 1)
+end
+TypeParameterAccessors.position(::Type{<:JLArray}, ::typeof(eltype)) = Position(1)
+TypeParameterAccessors.position(::Type{<:JLArray}, ::typeof(ndims)) = Position(2)