API for JIT modules - Githubissues

gorgonia / cu

package cu provides an idiomatic interface to the CUDA Driver API.

Apache License 2.0

475 stars 64 forks source link

API for JIT modules #13

Closed chewxy closed 6 years ago

chewxy commented 7 years ago

The C declaration is CUresult cuLinkCreate(unsigned int numOptions, CUjit_option* options, void** optionValues, CUlinkState* stateOut);

This requires the creation of new types, as prototyped below:

type Link uintptr (note that uintptr is just a placeholder. It's only good to use if the underlying type is a pointer, refer to cuda.h)
type JITOption int (with const)
type JITInputType byte (with const)

The APIs that need to be ported:

cuLinkCreate -> func NewLink() (Link, error)
cuLinkComplete -> func (Link) Complete() (fatBin unsafe.Pointer, size int64, err error)
cuLinkAddFile -> func (Link) AddFile(...) error
cuLinkAddData -> func (Link) AddData(...) error

chewxy commented 6 years ago

Fixed