Add cuda linspace and unittest.

2023-12-01 16:15:51 +08:00
parent 7db741502e
commit 65c78cd878
3 changed files with 32 additions and 0 deletions
--- a/src/Function1D.cuh
+++ b/src/Function1D.cuh
@@ -71,6 +71,8 @@ namespace Aurora

    CudaMatrix vecnorm(const CudaMatrix& aMatrix, NormMethod aNormMethod, int aDim);

+    CudaMatrix linspaceCuda(float aStart, float aEnd, int aNum);
+
    // ------compareSet----------------------------------------------------