Skip to content

P
pytensor
提交 cc62c3a8 authored 作者: nouiz's avatar nouiz
浏览文件
操作

Merge pull request #523 from ynd/sp_sandbox

Added MulSD, AddSS optimizations and new blas function declerations to sparse sandbox
上级 e71f0cb8 1c06a20a
全部展开 隐藏空白字符变更
内嵌 并排
正在显示 包含 2 行增加 和 0 行删除
theano/tensor/blas_headers.py
浏览文件 @ cc62c3a8
...@@ -604,6 +604,7 @@ def blas_header_text(): ...@@ -604,6 +604,7 @@ def blas_header_text():
void sswap_( const int*, float *, const int*, float *, const int*); void sswap_( const int*, float *, const int*, float *, const int*);
void scopy_( const int*, const float *, const int*, float *, const int*); void scopy_( const int*, const float *, const int*, float *, const int*);
void saxpy_( const int*, const float *, const float *, const int*, float *, const int*); void saxpy_( const int*, const float *, const float *, const int*, float *, const int*);
float sdot_(const int*, const float *, const int*, const float *, const int*);
void sdot_sub_(const int*, const float *, const int*, const float *, const int*, float *); void sdot_sub_(const int*, const float *, const int*, const float *, const int*, float *);
void sdsdot_sub_( const int*, const float *, const float *, const int*, const float *, const int*, float *); void sdsdot_sub_( const int*, const float *, const float *, const int*, const float *, const int*, float *);
void sscal_( const int*, const float *, float *, const int*); void sscal_( const int*, const float *, float *, const int*);
...@@ -621,6 +622,7 @@ def blas_header_text(): ...@@ -621,6 +622,7 @@ def blas_header_text():
void dcopy_( const int*, const double *, const int*, double *, const int*); void dcopy_( const int*, const double *, const int*, double *, const int*);
void daxpy_( const int*, const double *, const double *, const int*, double *, const int*); void daxpy_( const int*, const double *, const double *, const int*, double *, const int*);
void dswap_( const int*, double *, const int*, double *, const int*); void dswap_( const int*, double *, const int*, double *, const int*);
double ddot_(const int*, const double *, const int*, const double *, const int*);
void dsdot_sub_(const int*, const float *, const int*, const float *, const int*, double *); void dsdot_sub_(const int*, const float *, const int*, const float *, const int*, double *);
void ddot_sub_( const int*, const double *, const int*, const double *, const int*, double *); void ddot_sub_( const int*, const double *, const int*, const double *, const int*, double *);
void dscal_( const int*, const double *, double *, const int*); void dscal_( const int*, const double *, double *, const int*);
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论