Skip to content

Instantly share code, notes, and snippets.

@xrq-phys

xrq-phys/contract_fwd_prototype.jl

Last active July 25, 2020 15:08
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save xrq-phys/a90e3692a235f7c61c985df4039f6ff1 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save xrq-phys/a90e3692a235f7c61c985df4039f6ff1 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. xrq-phys renamed this gist Jul 25, 2020. 1 changed file with 0 additions and 0 deletions.
    0 contract_fd_prototype.jl → contract_fwd_prototype.jl
    View file
    File renamed without changes.
  2. xrq-phys revised this gist Jul 24, 2020. No changes.
  3. xrq-phys revised this gist Jul 24, 2020. 1 changed file with 7 additions and 0 deletions.
    7 changes: 7 additions & 0 deletions contract_fd_prototype.jl
    View file
    Original file line number Diff line number Diff line change
    @@ -1,3 +1,10 @@
    # here idx has 3 entries corresponding to e.g. "ik,jk->ij".
    contract!(A::Array{T},
    B::Array{T},
    C::Array{T},
    idx) where {T<:Dual} = contract!(T, sizeof(T)/sizeof(tovalue(T)), # tovalue unveils base type.
    A, 0, B, 0, C, 0, idx)

    contract!(Type::Dual{Tg, T, ND}, topst, # top-level stride
    A::Array, sftA, # arrays here are all at their top-level (not dispatched)
    B::Array, sftB,
  4. xrq-phys renamed this gist Jul 24, 2020. 1 changed file with 0 additions and 0 deletions.
    0 contract_prototype.jl → contract_fd_prototype.jl
    View file
    File renamed without changes.
  5. xrq-phys created this gist Jul 24, 2020.
    39 changes: 39 additions & 0 deletions contract_prototype.jl
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,39 @@
    contract!(Type::Dual{Tg, T, ND}, topst, # top-level stride
    A::Array, sftA, # arrays here are all at their top-level (not dispatched)
    B::Array, sftB,
    C::Array, sftC,
    idx) = begin
    # direct dispatch for value types.
    contract!(T, topst, A, sftA, B, sftB, C, sftC, idx)

    # for all differentials
    # TODO: consider exchangability
    for id = 1:ND
    # unpacks one layer of dual. note that differentials are also in value's type.
    contract!(T, topst,
    A, sftA + id*sizeof(T),
    B, sftB,
    C, sftC + id*sizeof(T),
    idx)
    contract!(T, topst,
    A, sftA,
    B, sftB + id*sizeof(T),
    C, sftC + id*sizeof(T),
    idx)
    end
    end

    contract!(Type::ValueType, # to be defined
    topst,
    A::Ptr{Cvoid}, sftA,
    B::Ptr{Cvoid}, sftB,
    C::Ptr{Cvoid}, sftC,
    idx) = begin
    # convert stride unit in top duals.
    stA = topst .* strides(A)
    stB = topst .* strides(B)
    stC = topst .* strides(C)
    # - build TBLIS-object for A, B and C from bare memory shifted by sft{A,B,C}.
    # - TLIBS-contract according to idx.
    return nothing
    end