Regenerate wrappers for 13.0.2 #2949
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Your PR requires formatting changes to meet the project's style guidelines. Click here to view the suggested changes.diff --git a/lib/cublas/libcublas.jl b/lib/cublas/libcublas.jl
index 7cd1d3d9e..18a76c44a 100644
--- a/lib/cublas/libcublas.jl
+++ b/lib/cublas/libcublas.jl
@@ -3138,66 +3138,94 @@ end
@checked function cublasGetEmulationSpecialValuesSupport(handle, mask)
initialize_context()
- @gcsafe_ccall libcublas.cublasGetEmulationSpecialValuesSupport(handle::cublasHandle_t,
- mask::Ptr{cudaEmulationSpecialValuesSupport})::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasGetEmulationSpecialValuesSupport(
+ handle::cublasHandle_t,
+ mask::Ptr{cudaEmulationSpecialValuesSupport}
+ )::cublasStatus_t
end
@checked function cublasSetEmulationSpecialValuesSupport(handle, mask)
initialize_context()
- @gcsafe_ccall libcublas.cublasSetEmulationSpecialValuesSupport(handle::cublasHandle_t,
- mask::cudaEmulationSpecialValuesSupport)::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasSetEmulationSpecialValuesSupport(
+ handle::cublasHandle_t,
+ mask::cudaEmulationSpecialValuesSupport
+ )::cublasStatus_t
end
@checked function cublasGetFixedPointEmulationMantissaControl(handle, mantissaControl)
initialize_context()
- @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMantissaControl(handle::cublasHandle_t,
- mantissaControl::Ptr{cudaEmulationMantissaControl})::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMantissaControl(
+ handle::cublasHandle_t,
+ mantissaControl::Ptr{cudaEmulationMantissaControl}
+ )::cublasStatus_t
end
@checked function cublasSetFixedPointEmulationMantissaControl(handle, mantissaControl)
initialize_context()
- @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMantissaControl(handle::cublasHandle_t,
- mantissaControl::cudaEmulationMantissaControl)::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMantissaControl(
+ handle::cublasHandle_t,
+ mantissaControl::cudaEmulationMantissaControl
+ )::cublasStatus_t
end
-@checked function cublasGetFixedPointEmulationMaxMantissaBitCount(handle,
- maxMantissaBitCount)
+@checked function cublasGetFixedPointEmulationMaxMantissaBitCount(
+ handle,
+ maxMantissaBitCount
+ )
initialize_context()
- @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMaxMantissaBitCount(handle::cublasHandle_t,
- maxMantissaBitCount::Ptr{Cint})::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMaxMantissaBitCount(
+ handle::cublasHandle_t,
+ maxMantissaBitCount::Ptr{Cint}
+ )::cublasStatus_t
end
-@checked function cublasSetFixedPointEmulationMaxMantissaBitCount(handle,
- maxMantissaBitCount)
+@checked function cublasSetFixedPointEmulationMaxMantissaBitCount(
+ handle,
+ maxMantissaBitCount
+ )
initialize_context()
- @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMaxMantissaBitCount(handle::cublasHandle_t,
- maxMantissaBitCount::Cint)::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMaxMantissaBitCount(
+ handle::cublasHandle_t,
+ maxMantissaBitCount::Cint
+ )::cublasStatus_t
end
@checked function cublasGetFixedPointEmulationMantissaBitOffset(handle, mantissaBitOffset)
initialize_context()
- @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMantissaBitOffset(handle::cublasHandle_t,
- mantissaBitOffset::Ptr{Cint})::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMantissaBitOffset(
+ handle::cublasHandle_t,
+ mantissaBitOffset::Ptr{Cint}
+ )::cublasStatus_t
end
@checked function cublasSetFixedPointEmulationMantissaBitOffset(handle, mantissaBitOffset)
initialize_context()
- @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMantissaBitOffset(handle::cublasHandle_t,
- mantissaBitOffset::Cint)::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMantissaBitOffset(
+ handle::cublasHandle_t,
+ mantissaBitOffset::Cint
+ )::cublasStatus_t
end
-@checked function cublasGetFixedPointEmulationMantissaBitCountPointer(handle,
- mantissaBitCount)
+@checked function cublasGetFixedPointEmulationMantissaBitCountPointer(
+ handle,
+ mantissaBitCount
+ )
initialize_context()
- @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMantissaBitCountPointer(handle::cublasHandle_t,
- mantissaBitCount::Ptr{Ptr{Cint}})::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasGetFixedPointEmulationMantissaBitCountPointer(
+ handle::cublasHandle_t,
+ mantissaBitCount::Ptr{Ptr{Cint}}
+ )::cublasStatus_t
end
-@checked function cublasSetFixedPointEmulationMantissaBitCountPointer(handle,
- mantissaBitCount)
+@checked function cublasSetFixedPointEmulationMantissaBitCountPointer(
+ handle,
+ mantissaBitCount
+ )
initialize_context()
- @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMantissaBitCountPointer(handle::cublasHandle_t,
- mantissaBitCount::Ptr{Cint})::cublasStatus_t
+ @gcsafe_ccall libcublas.cublasSetFixedPointEmulationMantissaBitCountPointer(
+ handle::cublasHandle_t,
+ mantissaBitCount::Ptr{Cint}
+ )::cublasStatus_t
end
function cublasGetStatusName(status)
diff --git a/lib/cublas/libcublasLt.jl b/lib/cublas/libcublasLt.jl
index 9abacf1e3..7f61afc48 100644
--- a/lib/cublas/libcublasLt.jl
+++ b/lib/cublas/libcublasLt.jl
@@ -87,7 +87,7 @@ end
const cublasLtMatmulPreference_t = Ptr{cublasLtMatmulPreferenceOpaque_t}
struct cublasLtEmulationDescOpaque_t
- data::NTuple{8,UInt64}
+ data::NTuple{8, UInt64}
end
const cublasLtEmulationDesc_t = Ptr{cublasLtEmulationDescOpaque_t}
@@ -1113,8 +1113,10 @@ end
@checked function cublasLtEmulationDescInit_internal(emulationDesc, size)
initialize_context()
- @gcsafe_ccall libcublasLt.cublasLtEmulationDescInit_internal(emulationDesc::cublasLtEmulationDesc_t,
- size::Csize_t)::cublasStatus_t
+ @gcsafe_ccall libcublasLt.cublasLtEmulationDescInit_internal(
+ emulationDesc::cublasLtEmulationDesc_t,
+ size::Csize_t
+ )::cublasStatus_t
end
@checked function cublasLtEmulationDescInit(emulationDesc)
@@ -1134,20 +1136,26 @@ end
@checked function cublasLtEmulationDescSetAttribute(emulationDesc, attr, buf, sizeInBytes)
initialize_context()
- @gcsafe_ccall libcublasLt.cublasLtEmulationDescSetAttribute(emulationDesc::cublasLtEmulationDesc_t,
- attr::cublasLtEmulationDescAttributes_t,
- buf::Ptr{Cvoid},
- sizeInBytes::Csize_t)::cublasStatus_t
-end
-
-@checked function cublasLtEmulationDescGetAttribute(emulationDesc, attr, buf, sizeInBytes,
- sizeWritten)
+ @gcsafe_ccall libcublasLt.cublasLtEmulationDescSetAttribute(
+ emulationDesc::cublasLtEmulationDesc_t,
+ attr::cublasLtEmulationDescAttributes_t,
+ buf::Ptr{Cvoid},
+ sizeInBytes::Csize_t
+ )::cublasStatus_t
+end
+
+@checked function cublasLtEmulationDescGetAttribute(
+ emulationDesc, attr, buf, sizeInBytes,
+ sizeWritten
+ )
initialize_context()
- @gcsafe_ccall libcublasLt.cublasLtEmulationDescGetAttribute(emulationDesc::cublasLtEmulationDesc_t,
- attr::cublasLtEmulationDescAttributes_t,
- buf::Ptr{Cvoid},
- sizeInBytes::Csize_t,
- sizeWritten::Ptr{Csize_t})::cublasStatus_t
+ @gcsafe_ccall libcublasLt.cublasLtEmulationDescGetAttribute(
+ emulationDesc::cublasLtEmulationDesc_t,
+ attr::cublasLtEmulationDescAttributes_t,
+ buf::Ptr{Cvoid},
+ sizeInBytes::Csize_t,
+ sizeWritten::Ptr{Csize_t}
+ )::cublasStatus_t
end
@cenum cublasLtReductionScheme_t::UInt32 begin
diff --git a/lib/cudadrv/libcuda.jl b/lib/cudadrv/libcuda.jl
index 0eba6a017..b3776548d 100644
--- a/lib/cudadrv/libcuda.jl
+++ b/lib/cudadrv/libcuda.jl
@@ -185,7 +185,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#298"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#298", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#298", private::Bool = false)
return (:smCount, if private
fieldnames(typeof(x))
else
@@ -592,7 +592,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#310"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#310", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#310", private::Bool = false)
return (:ptr, :array, if private
fieldnames(typeof(x))
else
@@ -1622,7 +1622,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#284"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#284", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#284", private::Bool = false)
return (:overBudget, if private
fieldnames(typeof(x))
else
@@ -2573,7 +2573,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#303"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#303", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#303", private::Bool = false)
return (:array, :mipmap, :linear, :pitch2D, :reserved, if private
fieldnames(typeof(x))
else
@@ -2852,7 +2852,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#286"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#286", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#286", private::Bool = false)
return (:fd, :win32, :nvSciBufObject, if private
fieldnames(typeof(x))
else
@@ -2955,7 +2955,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#277"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#277", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#277", private::Bool = false)
return (:fd, :win32, :nvSciSyncObj, if private
fieldnames(typeof(x))
else
@@ -3038,7 +3038,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#301"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#301", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#301", private::Bool = false)
return (:fence, :reserved, if private
fieldnames(typeof(x))
else
@@ -3088,7 +3088,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#299"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#299", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#299", private::Bool = false)
return (:fence, :nvSciSync, :keyedMutex, :reserved, if private
fieldnames(typeof(x))
else
@@ -3171,7 +3171,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#290"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#290", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#290", private::Bool = false)
return (:fence, :reserved, if private
fieldnames(typeof(x))
else
@@ -3223,7 +3223,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#288"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#288", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#288", private::Bool = false)
return (:fence, :nvSciSync, :keyedMutex, :reserved, if private
fieldnames(typeof(x))
else
@@ -3395,7 +3395,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#279"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#279", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#279", private::Bool = false)
return (:mipmap, :array, if private
fieldnames(typeof(x))
else
@@ -3424,7 +3424,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#280"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#280", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#280", private::Bool = false)
return (:sparseLevel, :miptail, if private
fieldnames(typeof(x))
else
@@ -3452,7 +3452,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#283"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#283", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#283", private::Bool = false)
return (:memHandle, if private
fieldnames(typeof(x))
else
diff --git a/lib/cupti/libcupti.jl b/lib/cupti/libcupti.jl
index 901b470a2..267ee5626 100644
--- a/lib/cupti/libcupti.jl
+++ b/lib/cupti/libcupti.jl
@@ -212,7 +212,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#425"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#425", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#425", private::Bool = false)
return (:stream, if private
fieldnames(typeof(x))
else
@@ -1654,7 +1654,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#354"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#354", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#354", private::Bool = false)
return (:size, :processId, if private
fieldnames(typeof(x))
else
@@ -1687,7 +1687,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#353"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#353", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#353", private::Bool = false)
return (:memoryPoolType, :pad2, :address, :releaseThreshold, :pool, :utilizedSize,
if private
fieldnames(typeof(x))
@@ -1820,7 +1820,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#391"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#391", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#391", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -1936,7 +1936,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#351"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#351", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#351", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -2416,7 +2416,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#411"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#411", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#411", private::Bool = false)
return (:cu, :cupti, if private
fieldnames(typeof(x))
else
@@ -2448,7 +2448,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#412"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#412", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#412", private::Bool = false)
return (:vDouble, :vUint32, :vUint64, :vInt32, :vInt64, if private
fieldnames(typeof(x))
else
@@ -2694,7 +2694,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#400"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#400", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#400", private::Bool = false)
return (:speed, :temperature, :power, :cooling, if private
fieldnames(typeof(x))
else
@@ -3528,7 +3528,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#393"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#393", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#393", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
@@ -3557,7 +3557,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#395"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#395", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#395", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
@@ -3634,7 +3634,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#405"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#405", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#405", private::Bool = false)
return (:devId, :bridgeId, if private
fieldnames(typeof(x))
else
@@ -3663,7 +3663,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#406"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#406", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#406", private::Bool = false)
return (:gpuAttr, :bridgeAttr, if private
fieldnames(typeof(x))
else
@@ -4700,7 +4700,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#409"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#409", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#409", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -4784,7 +4784,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#332"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#332", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#332", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -4873,7 +4873,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#355"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#355", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#355", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -4971,7 +4971,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#389"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#389", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#389", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -5072,7 +5072,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#336"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#336", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#336", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -5174,7 +5174,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#339"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#339", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#339", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -5279,7 +5279,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#423"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#423", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#423", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -5390,7 +5390,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#398"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#398", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#398", private::Bool = false)
return (:both, :config, if private
fieldnames(typeof(x))
else
@@ -5914,7 +5914,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#376"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#376", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#376", private::Bool = false)
return (:size, :processId, if private
fieldnames(typeof(x))
else
@@ -5946,7 +5946,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#375"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#375", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#375", private::Bool = false)
return (:memoryPoolType, :pad2, :address, :releaseThreshold, :pool,
if private
fieldnames(typeof(x))
@@ -6021,7 +6021,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#360"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#360", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#360", private::Bool = false)
return (:size, :processId, if private
fieldnames(typeof(x))
else
@@ -6054,7 +6054,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#359"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#359", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#359", private::Bool = false)
return (:memoryPoolType, :pad2, :address, :releaseThreshold, :pool, :utilizedSize,
if private
fieldnames(typeof(x))
@@ -6509,7 +6509,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#320"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#320", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#320", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
@@ -6538,7 +6538,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#322"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#322", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#322", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
@@ -6607,7 +6607,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#347"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#347", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#347", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
@@ -6636,7 +6636,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#349"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#349", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#349", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
@@ -6705,7 +6705,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#341"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#341", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#341", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
@@ -6734,7 +6734,7 @@ function Base.setproperty!(x::Ptr{var"##Ctag#343"}, f::Symbol, v)
return unsafe_store!(getproperty(x, f), v)
end
-function Base.propertynames(x::var"##Ctag#343", private::Bool=false)
+function Base.propertynames(x::var"##Ctag#343", private::Bool = false)
return (:uuidDev, :npu, if private
fieldnames(typeof(x))
else
diff --git a/lib/cutensor/src/libcutensor.jl b/lib/cutensor/src/libcutensor.jl
index 3650c4432..6f00624d5 100644
--- a/lib/cutensor/src/libcutensor.jl
+++ b/lib/cutensor/src/libcutensor.jl
@@ -214,7 +214,7 @@ end
numModes::UInt32,
extent::Ptr{Int64},
stride::Ptr{Int64},
- dataType::cudaDataType_t,
+ dataType::cudaDataType_t,
alignmentRequirement::UInt32)::cutensorStatus_t
end
@@ -488,21 +488,25 @@ end
stream::cudaStream_t)::cutensorStatus_t
end
-@checked function cutensorCreateBlockSparseTensorDescriptor(handle, desc, numModes,
- numNonZeroBlocks,
- numSectionsPerMode, extent,
- nonZeroCoordinates, stride,
- dataType)
+@checked function cutensorCreateBlockSparseTensorDescriptor(
+ handle, desc, numModes,
+ numNonZeroBlocks,
+ numSectionsPerMode, extent,
+ nonZeroCoordinates, stride,
+ dataType
+ )
initialize_context()
- @gcsafe_ccall libcutensor.cutensorCreateBlockSparseTensorDescriptor(handle::cutensorHandle_t,
- desc::Ptr{cutensorBlockSparseTensorDescriptor_t},
- numModes::UInt32,
- numNonZeroBlocks::UInt64,
- numSectionsPerMode::Ptr{UInt32},
- extent::Ptr{Int64},
- nonZeroCoordinates::Ptr{Int32},
- stride::Ptr{Int64},
- dataType::cudaDataType_t)::cutensorStatus_t
+ @gcsafe_ccall libcutensor.cutensorCreateBlockSparseTensorDescriptor(
+ handle::cutensorHandle_t,
+ desc::Ptr{cutensorBlockSparseTensorDescriptor_t},
+ numModes::UInt32,
+ numNonZeroBlocks::UInt64,
+ numSectionsPerMode::Ptr{UInt32},
+ extent::Ptr{Int64},
+ nonZeroCoordinates::Ptr{Int32},
+ stride::Ptr{Int64},
+ dataType::cudaDataType_t
+ )::cutensorStatus_t
end
@checked function cutensorDestroyBlockSparseTensorDescriptor(desc)
@@ -510,40 +514,48 @@ end
@gcsafe_ccall libcutensor.cutensorDestroyBlockSparseTensorDescriptor(desc::cutensorBlockSparseTensorDescriptor_t)::cutensorStatus_t
end
-@checked function cutensorCreateBlockSparseContraction(handle, desc, descA, modeA, opA,
- descB, modeB, opB, descC, modeC, opC,
- descD, modeD, descCompute)
- initialize_context()
- @gcsafe_ccall libcutensor.cutensorCreateBlockSparseContraction(handle::cutensorHandle_t,
- desc::Ptr{cutensorOperationDescriptor_t},
- descA::cutensorBlockSparseTensorDescriptor_t,
- modeA::Ptr{Int32},
- opA::cutensorOperator_t,
- descB::cutensorBlockSparseTensorDescriptor_t,
- modeB::Ptr{Int32},
- opB::cutensorOperator_t,
- descC::cutensorBlockSparseTensorDescriptor_t,
- modeC::Ptr{Int32},
- opC::cutensorOperator_t,
- descD::cutensorBlockSparseTensorDescriptor_t,
- modeD::Ptr{Int32},
- descCompute::cutensorComputeDescriptor_t)::cutensorStatus_t
-end
-
-@checked function cutensorBlockSparseContract(handle, plan, alpha, A, B, beta, C, D,
- workspace, workspaceSize, stream)
- initialize_context()
- @gcsafe_ccall libcutensor.cutensorBlockSparseContract(handle::cutensorHandle_t,
- plan::cutensorPlan_t,
- alpha::Ptr{Cvoid},
- A::Ptr{Ptr{Cvoid}},
- B::Ptr{Ptr{Cvoid}},
- beta::Ptr{Cvoid},
- C::Ptr{Ptr{Cvoid}},
- D::Ptr{Ptr{Cvoid}},
- workspace::Ptr{Cvoid},
- workspaceSize::UInt64,
- stream::cudaStream_t)::cutensorStatus_t
+@checked function cutensorCreateBlockSparseContraction(
+ handle, desc, descA, modeA, opA,
+ descB, modeB, opB, descC, modeC, opC,
+ descD, modeD, descCompute
+ )
+ initialize_context()
+ @gcsafe_ccall libcutensor.cutensorCreateBlockSparseContraction(
+ handle::cutensorHandle_t,
+ desc::Ptr{cutensorOperationDescriptor_t},
+ descA::cutensorBlockSparseTensorDescriptor_t,
+ modeA::Ptr{Int32},
+ opA::cutensorOperator_t,
+ descB::cutensorBlockSparseTensorDescriptor_t,
+ modeB::Ptr{Int32},
+ opB::cutensorOperator_t,
+ descC::cutensorBlockSparseTensorDescriptor_t,
+ modeC::Ptr{Int32},
+ opC::cutensorOperator_t,
+ descD::cutensorBlockSparseTensorDescriptor_t,
+ modeD::Ptr{Int32},
+ descCompute::cutensorComputeDescriptor_t
+ )::cutensorStatus_t
+end
+
+@checked function cutensorBlockSparseContract(
+ handle, plan, alpha, A, B, beta, C, D,
+ workspace, workspaceSize, stream
+ )
+ initialize_context()
+ @gcsafe_ccall libcutensor.cutensorBlockSparseContract(
+ handle::cutensorHandle_t,
+ plan::cutensorPlan_t,
+ alpha::Ptr{Cvoid},
+ A::Ptr{Ptr{Cvoid}},
+ B::Ptr{Ptr{Cvoid}},
+ beta::Ptr{Cvoid},
+ C::Ptr{Ptr{Cvoid}},
+ D::Ptr{Ptr{Cvoid}},
+ workspace::Ptr{Cvoid},
+ workspaceSize::UInt64,
+ stream::cudaStream_t
+ )::cutensorStatus_t
end
function cutensorGetErrorString(error)
diff --git a/lib/cutensornet/src/libcutensornet.jl b/lib/cutensornet/src/libcutensornet.jl
index f45b193d3..e6d84796b 100644
--- a/lib/cutensornet/src/libcutensornet.jl
+++ b/lib/cutensornet/src/libcutensornet.jl
@@ -422,7 +422,8 @@ end
extentsIn, stridesIn, modesIn,
qualifiersIn, numModesOut, extentsOut,
stridesOut, modesOut, dataType,
- computeType, networkDesc)
+ computeType, networkDesc
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetCreateNetworkDescriptor(handle::cutensornetHandle_t,
numInputs::Int32,
@@ -437,13 +438,16 @@ end
modesOut::Ptr{Int32},
dataType::cudaDataType_t,
computeType::cutensornetComputeType_t,
- networkDesc::Ptr{cutensornetNetworkDescriptor_t})::cutensornetStatus_t
+ networkDesc::Ptr{cutensornetNetworkDescriptor_t}
+ )::cutensornetStatus_t
end
@checked function cutensornetCreateNetwork(handle, networkDesc)
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetCreateNetwork(handle::cutensornetHandle_t,
- networkDesc::Ptr{cutensornetNetworkDescriptor_t})::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetCreateNetwork(
+ handle::cutensornetHandle_t,
+ networkDesc::Ptr{cutensornetNetworkDescriptor_t}
+ )::cutensornetStatus_t
end
@checked function cutensornetDestroyNetworkDescriptor(networkDesc)
@@ -456,54 +460,65 @@ end
@gcsafe_ccall libcutensornet.cutensornetDestroyNetwork(networkDesc::cutensornetNetworkDescriptor_t)::cutensornetStatus_t
end
-@checked function cutensornetNetworkAppendTensor(handle, networkDesc, numModes, extents,
- modeLabels, qualifiers, dataType, tensorId)
+@checked function cutensornetNetworkAppendTensor(
+ handle, networkDesc, numModes, extents,
+ modeLabels, qualifiers, dataType, tensorId
+ )
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkAppendTensor(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- numModes::Int32,
- extents::Ptr{Int64},
- modeLabels::Ptr{Int32},
- qualifiers::Ptr{cutensornetTensorQualifiers_t},
- dataType::cudaDataType_t,
- tensorId::Ptr{Int64})::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetNetworkAppendTensor(
+ handle::cutensornetHandle_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
+ numModes::Int32,
+ extents::Ptr{Int64},
+ modeLabels::Ptr{Int32},
+ qualifiers::Ptr{cutensornetTensorQualifiers_t},
+ dataType::cudaDataType_t,
+ tensorId::Ptr{Int64}
+ )::cutensornetStatus_t
end
-@checked function cutensornetNetworkSetOutputTensor(handle, networkDesc, numModes,
- modeLabels, dataType)
+@checked function cutensornetNetworkSetOutputTensor(
+ handle, networkDesc, numModes,
+ modeLabels, dataType
+ )
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkSetOutputTensor(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- numModes::Int32,
- modeLabels::Ptr{Int32},
- dataType::cudaDataType_t)::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetNetworkSetOutputTensor(
+ handle::cutensornetHandle_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
+ numModes::Int32,
+ modeLabels::Ptr{Int32},
+ dataType::cudaDataType_t
+ )::cutensornetStatus_t
end
-@checked function cutensornetNetworkGetAttribute(handle, networkDesc, attr, buffer,
+@checked function cutensornetNetworkGetAttribute(
+ handle, networkDesc, attr, buffer,
sizeInBytes)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetNetworkGetAttribute(handle::cutensornetHandle_t,
networkDesc::cutensornetNetworkDescriptor_t,
attr::cutensornetNetworkAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
-@checked function cutensornetNetworkSetAttribute(handle, networkDesc, attr, buffer,
+@checked function cutensornetNetworkSetAttribute(
+ handle, networkDesc, attr, buffer,
sizeInBytes)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetNetworkSetAttribute(handle::cutensornetHandle_t,
networkDesc::cutensornetNetworkDescriptor_t,
attr::cutensornetNetworkAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
-@checked function cutensornetGetOutputTensorDetails(handle, networkDesc, numModes, dataSize,
+@checked function cutensornetGetOutputTensorDetails(
+ handle, networkDesc, numModes, dataSize,
modeLabels, extents, strides)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetGetOutputTensorDetails(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
numModes::Ptr{Int32},
dataSize::Ptr{Csize_t},
modeLabels::Ptr{Int32},
@@ -511,11 +526,13 @@ end
strides::Ptr{Int64})::cutensornetStatus_t
end
-@checked function cutensornetGetOutputTensorDescriptor(handle, networkDesc,
- outputTensorDesc)
+@checked function cutensornetGetOutputTensorDescriptor(
+ handle, networkDesc,
+ outputTensorDesc
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetGetOutputTensorDescriptor(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
outputTensorDesc::Ptr{cutensornetTensorDescriptor_t})::cutensornetStatus_t
end
@@ -537,20 +554,23 @@ end
workDesc::Ptr{cutensornetWorkspaceDescriptor_t})::cutensornetStatus_t
end
-@checked function cutensornetWorkspaceComputeSizes(handle, networkDesc, optimizerInfo,
- workDesc)
+@checked function cutensornetWorkspaceComputeSizes(
+ handle, networkDesc, optimizerInfo,
+ workDesc
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetWorkspaceComputeSizes(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
optimizerInfo::cutensornetContractionOptimizerInfo_t,
workDesc::cutensornetWorkspaceDescriptor_t)::cutensornetStatus_t
end
-@checked function cutensornetWorkspaceComputeContractionSizes(handle, networkDesc,
+@checked function cutensornetWorkspaceComputeContractionSizes(
+ handle, networkDesc,
optimizerInfo, workDesc)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetWorkspaceComputeContractionSizes(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
optimizerInfo::cutensornetContractionOptimizerInfo_t,
workDesc::cutensornetWorkspaceDescriptor_t)::cutensornetStatus_t
end
@@ -642,24 +662,26 @@ end
end
@checked function cutensornetContractionOptimizerConfigGetAttribute(handle, optimizerConfig,
- attr, buffer,
- sizeInBytes)
+ attr, buffer,
+ sizeInBytes
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetContractionOptimizerConfigGetAttribute(handle::cutensornetHandle_t,
optimizerConfig::cutensornetContractionOptimizerConfig_t,
attr::cutensornetContractionOptimizerConfigAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
@checked function cutensornetContractionOptimizerConfigSetAttribute(handle, optimizerConfig,
- attr, buffer,
- sizeInBytes)
+ attr, buffer,
+ sizeInBytes
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetContractionOptimizerConfigSetAttribute(handle::cutensornetHandle_t,
optimizerConfig::cutensornetContractionOptimizerConfig_t,
attr::cutensornetContractionOptimizerConfigAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
@@ -668,19 +690,22 @@ end
@gcsafe_ccall libcutensornet.cutensornetDestroyContractionOptimizerInfo(optimizerInfo::cutensornetContractionOptimizerInfo_t)::cutensornetStatus_t
end
-@checked function cutensornetCreateContractionOptimizerInfo(handle, networkDesc,
- optimizerInfo)
+@checked function cutensornetCreateContractionOptimizerInfo(
+ handle, networkDesc,
+ optimizerInfo
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetCreateContractionOptimizerInfo(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
optimizerInfo::Ptr{cutensornetContractionOptimizerInfo_t})::cutensornetStatus_t
end
-@checked function cutensornetContractionOptimize(handle, networkDesc, optimizerConfig,
+@checked function cutensornetContractionOptimize(
+ handle, networkDesc, optimizerConfig,
workspaceSizeConstraint, optimizerInfo)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetContractionOptimize(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
optimizerConfig::cutensornetContractionOptimizerConfig_t,
workspaceSizeConstraint::UInt64,
optimizerInfo::cutensornetContractionOptimizerInfo_t)::cutensornetStatus_t
@@ -688,28 +713,32 @@ end
@checked function cutensornetNetworkSetOptimizerInfo(handle, networkDesc, optimizerInfo)
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkSetOptimizerInfo(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- optimizerInfo::cutensornetContractionOptimizerInfo_t)::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetNetworkSetOptimizerInfo(
+ handle::cutensornetHandle_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
+ optimizerInfo::cutensornetContractionOptimizerInfo_t
+ )::cutensornetStatus_t
end
@checked function cutensornetContractionOptimizerInfoGetAttribute(handle, optimizerInfo,
- attr, buffer, sizeInBytes)
+ attr, buffer, sizeInBytes
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetContractionOptimizerInfoGetAttribute(handle::cutensornetHandle_t,
optimizerInfo::cutensornetContractionOptimizerInfo_t,
attr::cutensornetContractionOptimizerInfoAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
@checked function cutensornetContractionOptimizerInfoSetAttribute(handle, optimizerInfo,
- attr, buffer, sizeInBytes)
+ attr, buffer, sizeInBytes
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetContractionOptimizerInfoSetAttribute(handle::cutensornetHandle_t,
optimizerInfo::cutensornetContractionOptimizerInfo_t,
attr::cutensornetContractionOptimizerInfoAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
@@ -730,14 +759,15 @@ end
sizeInBytes::Csize_t)::cutensornetStatus_t
end
-@checked function cutensornetCreateContractionOptimizerInfoFromPackedData(handle,
- networkDesc,
+@checked function cutensornetCreateContractionOptimizerInfoFromPackedData(
+ handle,
+ networkDesc,
buffer,
sizeInBytes,
optimizerInfo)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetCreateContractionOptimizerInfoFromPackedData(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
buffer::Ptr{Cvoid},
sizeInBytes::Csize_t,
optimizerInfo::Ptr{cutensornetContractionOptimizerInfo_t})::cutensornetStatus_t
@@ -753,11 +783,13 @@ end
optimizerInfo::cutensornetContractionOptimizerInfo_t)::cutensornetStatus_t
end
-@checked function cutensornetCreateContractionPlan(handle, networkDesc, optimizerInfo,
- workDesc, plan)
+@checked function cutensornetCreateContractionPlan(
+ handle, networkDesc, optimizerInfo,
+ workDesc, plan
+ )
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetCreateContractionPlan(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
optimizerInfo::cutensornetContractionOptimizerInfo_t,
workDesc::cutensornetWorkspaceDescriptor_t,
plan::Ptr{cutensornetContractionPlan_t})::cutensornetStatus_t
@@ -765,9 +797,11 @@ end
@checked function cutensornetNetworkPrepareContraction(handle, networkDesc, workDesc)
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkPrepareContraction(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- workDesc::cutensornetWorkspaceDescriptor_t)::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetNetworkPrepareContraction(
+ handle::cutensornetHandle_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
+ workDesc::cutensornetWorkspaceDescriptor_t
+ )::cutensornetStatus_t
end
@checked function cutensornetDestroyContractionPlan(plan)
@@ -787,14 +821,18 @@ end
stream::cudaStream_t)::cutensornetStatus_t
end
-@checked function cutensornetNetworkAutotuneContraction(handle, networkDesc, workDesc, pref,
- stream)
+@checked function cutensornetNetworkAutotuneContraction(
+ handle, networkDesc, workDesc, pref,
+ stream
+ )
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkAutotuneContraction(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- workDesc::cutensornetWorkspaceDescriptor_t,
- pref::cutensornetNetworkAutotunePreference_t,
- stream::cudaStream_t)::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetNetworkAutotuneContraction(
+ handle::cutensornetHandle_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
+ workDesc::cutensornetWorkspaceDescriptor_t,
+ pref::cutensornetNetworkAutotunePreference_t,
+ stream::cudaStream_t
+ )::cutensornetStatus_t
end
@checked function cutensornetCreateContractionAutotunePreference(handle, autotunePreference)
@@ -805,54 +843,64 @@ end
@checked function cutensornetCreateNetworkAutotunePreference(handle, autotunePreference)
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetCreateNetworkAutotunePreference(handle::cutensornetHandle_t,
- autotunePreference::Ptr{cutensornetNetworkAutotunePreference_t})::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetCreateNetworkAutotunePreference(
+ handle::cutensornetHandle_t,
+ autotunePreference::Ptr{cutensornetNetworkAutotunePreference_t}
+ )::cutensornetStatus_t
end
@checked function cutensornetContractionAutotunePreferenceGetAttribute(handle,
autotunePreference,
- attr, buffer,
+ attr, buffer,
sizeInBytes)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetContractionAutotunePreferenceGetAttribute(handle::cutensornetHandle_t,
autotunePreference::cutensornetContractionAutotunePreference_t,
attr::cutensornetContractionAutotunePreferenceAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
-@checked function cutensornetNetworkAutotunePreferenceGetAttribute(handle,
- autotunePreference, attr,
- buffer, sizeInBytes)
+@checked function cutensornetNetworkAutotunePreferenceGetAttribute(
+ handle,
+ autotunePreference, attr,
+ buffer, sizeInBytes
+ )
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkAutotunePreferenceGetAttribute(handle::cutensornetHandle_t,
- autotunePreference::cutensornetNetworkAutotunePreference_t,
- attr::cutensornetNetworkAutotunePreferenceAttributes_t,
- buffer::Ptr{Cvoid},
- sizeInBytes::Csize_t)::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetNetworkAutotunePreferenceGetAttribute(
+ handle::cutensornetHandle_t,
+ autotunePreference::cutensornetNetworkAutotunePreference_t,
+ attr::cutensornetNetworkAutotunePreferenceAttributes_t,
+ buffer::Ptr{Cvoid},
+ sizeInBytes::Csize_t
+ )::cutensornetStatus_t
end
@checked function cutensornetContractionAutotunePreferenceSetAttribute(handle,
autotunePreference,
- attr, buffer,
+ attr, buffer,
sizeInBytes)
initialize_context()
@gcsafe_ccall libcutensornet.cutensornetContractionAutotunePreferenceSetAttribute(handle::cutensornetHandle_t,
autotunePreference::cutensornetContractionAutotunePreference_t,
attr::cutensornetContractionAutotunePreferenceAttributes_t,
- buffer::Ptr{Cvoid},
+ buffer::Ptr{Cvoid},
sizeInBytes::Csize_t)::cutensornetStatus_t
end
-@checked function cutensornetNetworkAutotunePreferenceSetAttribute(handle,
- autotunePreference, attr,
- buf, sizeInBytes)
+@checked function cutensornetNetworkAutotunePreferenceSetAttribute(
+ handle,
+ autotunePreference, attr,
+ buf, sizeInBytes
+ )
initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkAutotunePreferenceSetAttribute(handle::cutensornetHandle_t,
- autotunePreference::cutensornetNetworkAutotunePreference_t,
- attr::cutensornetNetworkAutotunePreferenceAttributes_t,
- buf::Ptr{Cvoid},
- sizeInBytes::Csize_t)::cutensornetStatus_t
+ @gcsafe_ccall libcutensornet.cutensornetNetworkAutotunePreferenceSetAttribute(
+ handle::cutensornetHandle_t,
+ autotunePreference::cutensornetNetworkAutotunePreference_t,
+ attr::cutensornetNetworkAutotunePreferenceAttributes_t,
+ buf::Ptr{Cvoid},
+ sizeInBytes::Csize_t
+ )::cutensornetStatus_t
end
@checked function cutensornetDestroyContractionAutotunePreference(autotunePreference)
@@ -914,85 +962,113 @@ end
stream::cudaStream_t)::cutensornetStatus_t
end
-@checked function cutensornetNetworkSetInputTensorMemory(handle, networkDesc, tensorId,
- buffer, strides)
- initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkSetInputTensorMemory(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- tensorId::Int64,
- buffer::Ptr{Cvoid},
- strides::Ptr{Int64})::cutensornetStatus_t
-end
-
-@checked function cutensornetNetworkSetOutputTensorMemory(handle, networkDesc, buffer,
- strides)
- initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkSetOutputTensorMemory(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- buffer::Ptr{Cvoid},
- strides::Ptr{Int64})::cutensornetStatus_t
-end
-
-@checked function cutensornetNetworkSetGradientTensorMemory(handle, networkDesc,
- correspondingTensorId, buffer,
- strides)
- initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkSetGradientTensorMemory(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- correspondingTensorId::Int64,
- buffer::Ptr{Cvoid},
- strides::Ptr{Int64})::cutensornetStatus_t
-end
-
-@checked function cutensornetNetworkSetAdjointTensorMemory(handle, networkDesc, buffer,
- strides)
- initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkSetAdjointTensorMemory(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- buffer::Ptr{Cvoid},
- strides::Ptr{Int64})::cutensornetStatus_t
-end
-
-@checked function cutensornetNetworkContract(handle, networkDesc, accumulateOutput,
- workDesc, sliceGroup, stream)
- initialize_context()
- @gcsafe_ccall libcutensornet.cutensornetNetworkContract(handle::cutensornetHandle_t,
- networkDesc::cutensornetNetworkDescriptor_t,
- accumulateOutput::Int32,
- workDesc::cutensornetWorkspaceDescriptor_t,
- sliceGroup::cutensornetSliceGroup_t,
- stream::cudaStream_t)::cutensornetStatus_t
+@checked function cutensornetNetworkSetInputTensorMemory(
+ handle, networkDesc, tensorId,
+ buffer, strides
+ )
+ initialize_context()
+ @gcsafe_ccall libcutensornet.cutensornetNetworkSetInputTensorMemory(
+ handle::cutensornetHandle_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
+ tensorId::Int64,
+ buffer::Ptr{Cvoid},
+ strides::Ptr{Int64}
+ )::cutensornetStatus_t
+end
+
+@checked function cutensornetNetworkSetOutputTensorMemory(
+ handle, networkDesc, buffer,
+ strides
+ )
+ initialize_context()
+ @gcsafe_ccall libcutensornet.cutensornetNetworkSetOutputTensorMemory(
+ handle::cutensornetHandle_t,
+ networkDesc::cutensornetNetworkDescriptor_t,
+ buffer::Ptr{Cvoid},
+ strides::Ptr{Int64}
+ )::cutensornetStatus_t
+end
+
+@checked function cutensornetNetworkSetGradientTensorMemory(
+ handle, networkDesc,
+ correspondingTensorId, buffer,
+ ...*[Comment body truncated]* |
|
Needs some TLC for CUBLAS, I'll look at this today |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.