• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    迪恩网络公众号

JuliaGPU/Adapt.jl

原作者: [db:作者] 来自: 网络 收藏 邀请

开源软件名称:

JuliaGPU/Adapt.jl

开源软件地址:

https://github.com/JuliaGPU/Adapt.jl

开源编程语言:

Julia 100.0%

开源软件介绍:

Adapt

Build Status Coverage

The adapt(T, x) function acts like convert(T, x), but without the restriction of returning a T. This allows you to "convert" wrapper types like Adjoint to be GPU compatible (for example) without throwing away the wrapper.

For example:

adapt(CuArray, ::Adjoint{Array})::Adjoint{CuArray}

New wrapper types like Adjoint should overload adapt_structure(T, ::Adjoint) (usually just to forward the call to adapt):

Adapt.adapt_structure(to, x::Adjoint) = Adjoint(adapt(to, parent(x)))

A similar function, adapt_storage, can be used to define the conversion behavior for the innermost storage types:

adapt_storage(::Type{<:CuArray}, xs::AbstractArray) = convert(CuArray, xs)

Implementations of adapt_storage will typically be part of libraries that use Adapt. For example, CUDA.jl defines methods of adapt_storage(::Type{<:CuArray}, ...) and uses that to convert different kinds of arrays, while CUDAnative.jl provides implementations of adapt_storage(::CUDAnative.Adaptor, ...) to convert various values to GPU-compatible alternatives.

Packages that define new wrapper types and want to be compatible with packages that use Adapt.jl should provide implementations of adapt_structure that preserve the wrapper type. Adapt.jl already provides such methods for array wrappers that are part of the Julia standard library.




鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap