Add log function and fix start kernel block thread size.

2023-11-27 13:51:35 +08:00
parent 980865b494
commit 9d8c204ec4
2 changed files with 51 additions and 16 deletions
--- a/src/Function1D.cuh
+++ b/src/Function1D.cuh
@@ -45,6 +45,8 @@ namespace Aurora
    CudaMatrix repmat(const CudaMatrix& aMatrix,int aRowTimes, int aColumnTimes, int aSliceTimes);

    CudaMatrix repmat3d(const CudaMatrix& aMatrix,int aRowTimes, int aColumnTimes, int aSliceTimes);
+
+    CudaMatrix log(const CudaMatrix& aMatrix, int aBaseNum = -1);
 }

 #endif //AURORA_CUDA_FUNCTION1D_H