将参数从 Fortran 程序传递给 Fortran 函数

您可以从 Fortran MEX 文件中调用 LAPACK 和 BLAS 函数。以下示例使用两个矩阵,并通过调用 BLAS 例程 dgemm 将这两个矩阵相乘。要运行该示例,请将代码复制到编辑器中并将文件命名为 calldgemm.F。

#include "fintrf.h"

subroutine mexFunction(nlhs, plhs, nrhs, prhs)

mwPointer plhs(*), prhs(*)

integer nlhs, nrhs

mwPointer mxcreatedoublematrix

mwPointer mxgetpr

mwPointer A, B, C

mwSize mxgetm, mxgetn

mwSignedIndex m, n, p

mwSize numel

double precision one, zero, ar, br

character ch1, ch2

ch1 = 'N'

ch2 = 'N'

one = 1.0

zero = 0.0

A = mxgetpr(prhs(1))

B = mxgetpr(prhs(2))

m = mxgetm(prhs(1))

p = mxgetn(prhs(1))

n = mxgetn(prhs(2))

plhs(1) = mxcreatedoublematrix(m, n, 0.0)

C = mxgetpr(plhs(1))

numel = 1

call mxcopyptrtoreal8(A, ar, numel)

call mxcopyptrtoreal8(B, br, numel)

call dgemm(ch1, ch2, m, n, p, one, %val(A), m,

+ %val(B), p, zero, %val(C), m)

return

end

链接到 BLAS 库,其中包含 dgemm 函数。

mex -v -R2017b calldgemm.F -lmwblas

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐