Skip to content

[WIP] Rewriting the wrapper to use luaffi #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
pavanky wants to merge 9 commits into
base: master
Choose a base branch
from
Open

[WIP] Rewriting the wrapper to use luaffi #3

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
b6210b4
Refactoring the repo to use ffi
pavanky Nov 3, 2016
b0dcd76
Adding C definitions from device.h and array.h
pavanky Feb 8, 2017
1e96bcd
Adding enums from arrayfire
pavanky Feb 23, 2017
3efd323
Adding arrayfire.Array class and necessary helper functions
pavanky Feb 23, 2017
60e06eb
Adding better af.print and updating README
pavanky Feb 23, 2017
9cbe920
Adding functions from array.h
pavanky Feb 26, 2017
c923238
Adding airthmetic operations
pavanky Feb 27, 2017
bde6e26
Adding functions from algorithm.h
pavanky Feb 27, 2017
44bf5e6
Adding functions from blas.h
pavanky Feb 27, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Adding functions from blas.h
  • Loading branch information
@pavanky
pavanky committed Feb 27, 2017
commit 44bf5e6536b70d8fd73d8e6a37cfaddbdbab7cb8
1 change: 1 addition & 0 deletions arrayfire.lua
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ require('arrayfire.util')
require('arrayfire.array')
require('arrayfire.arith')
require('arrayfire.algorithm')
require('arrayfire.blas')
require('arrayfire.device')

return af
68 changes: 68 additions & 0 deletions arrayfire/blas.lua
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
require('arrayfire.lib')
require('arrayfire.defines')
require('arrayfire.array')
local ffi = require( "ffi" )

local funcs = {}

funcs[30] = [[
af_err af_matmul( af_array *out ,
const af_array lhs, const af_array rhs,
const af_mat_prop optLhs, const af_mat_prop optRhs);
af_err af_dot(af_array *out,
const af_array lhs, const af_array rhs,
const af_mat_prop optLhs, const af_mat_prop optRhs);
af_err af_transpose(af_array *out, af_array in, const bool conjugate);
af_err af_transpose_inplace(af_array in, const bool conjugate);
]]

funcs[35] = [[
af_err af_dot_all(double *real, double *imag,
const af_array lhs, const af_array rhs,
const af_mat_prop optLhs, const af_mat_prop optRhs);
]]

af.lib.cdef(funcs)
local c_array_p = af.ffi.c_array_p
local init = af.Array.init
local c_ptr_t = af.ffi.c_ptr_t

af.matmul = function(lhs, rhs, lhsOpts, rhsOpts)
local res = c_array_p()
af.clib.af_matmul(res, lhs:get(), rhs:get(),
lhsOpts or af.mat_prop.none,
rhsOpts or af.mat_prop.none)
return init(res[0])
end

af.dot = function(lhs, rhs, lhsOpts, rhsOpts)
local res = c_array_p()
af.clib.af_dot(res, lhs:get(), rhs:get(),
lhsOpts or af.mat_prop.none,
rhsOpts or af.mat_prop.none)
return init(res[0])
end

af.transpose = function(input, inplace, conjugate)
if inplace then
af.clib.af_transpose_inplace(input:get(), conjugate == nil and false or true)
return input
else
local res = c_array_p()
af.clib.af_transpose(res, input:get(), conjugate == nil and false or true)
return init(res[0])
end
end

af.ctranspose = function(input, inplace)
return af.transpose(input, inplace, true)
end

af.vecdot = function(lhs, rhs, lhsOpts, rhsOpts)
local res = c_ptr_t('double', 2)
af.clib.af_dot_all(res + 0, res + 1,
lhs:get(), rhs:get(),
lhsOpts or af.mat_prop.none,
rhsOpts or af.mat_prop.none)
return (res[1] == 0) and res[0] or {real = res[0], imag = res[1]}
end
1 change: 1 addition & 0 deletions rocks/arrayfire-scm-1.rockspec
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,5 +25,6 @@ build = {
["arrayfire.dim4"] = "arrayfire/dim4.lua",
["arrayfire.arith"] = "arrayfire/arith.lua",
["arrayfire.algorithm"] = "arrayfire/algorithm.lua",
["arrayfire.blas"] = "arrayfire/blas.lua",
},
}