mv tensor repo

2023-11-29 22:46:44 +01:00
parent 55852146cb
commit 09c402a4ba
7 changed files with 4 additions and 1 deletions
@@ -0,0 +1,31 @@
+#ifndef __TENS_CUDA_H__
+#define __TENS_CUDA_H__
+
+#include <cstdio>
+#include <cstdlib>
+
+#include <stdexcept>
+
+//#include "/home/fanasina/progr_/ptens0neD/tensor/tens0neD/tens0neD.h"
+#include "tensor/tens0neD/tens0neD.h"
+
+//#include "/home/fanasina/progr_/ptens0neD/tensor/tensCuda/d_tensCuda.h"
+#include "tensor/tensCuda/d_tensCuda.h"
+//#include "dimension/dimension.h"
+
+template<typename T>
+struct Tensor;
+
+template<typename T>
+void cudaTensorContractNestProd(Tensor<T>& M, const Tensor<T>& M0, const Tensor<T>& M1, int nestingDepth, bool strict = true);
+
+template<typename T>
+void cudaTensorProd(Tensor<T>& M, const Tensor<T>& M0, const Tensor<T>& M1);
+template<typename T>
+void cudaTensorProdEnd(Tensor<T>& M, const Tensor<T>& M0, const Tensor<T>& M1);
+template<typename T>
+void cudapermuteTensor(Tensor<T>& M, const Tensor<T>& M0, permutation p);
+
+
+#endif
+