Commit 4102c395 authored by Xuan Thang Nguyen's avatar Xuan Thang Nguyen
Browse files

refactoring

parent 08769c95

src/quicksort/cudaPartition.cuh

+9 −5
Original line number Diff line number Diff line
@@ -16,7 +16,7 @@ __device__ Value pickPivot(TNL::Containers::ArrayView<Value, Device> src, const 
    if (src.getSize() == 1)

        return src[0];



    Value a = src[0], b = src[src.getSize() / 2], c = src[src.getSize() - 1];

    const Value &a = src[0], &b = src[src.getSize() / 2], &c = src[src.getSize() - 1];



    if (Cmp(a, b)) // ..a..b..

    {
@@ -47,7 +47,7 @@ __device__ int pickPivotIdx(TNL::Containers::ArrayView<Value, Device> src, const 
    if (src.getSize() <= 1)

        return 0;



    Value a = src[0], b = src[src.getSize() / 2], c = src[src.getSize() - 1];

    const Value &a = src[0], &b = src[src.getSize() / 2], &c = src[src.getSize() - 1];



    if (Cmp(a, b)) // ..a..b..

    {
@@ -69,6 +69,8 @@ __device__ int pickPivotIdx(TNL::Containers::ArrayView<Value, Device> src, const 
    }

}



//-----------------------------------------------------------



template <typename Value, typename Function>

__device__ void countElem(ArrayView<Value, Devices::Cuda> arr,

                          const Function &Cmp,
@@ -77,7 +79,7 @@ __device__ void countElem(ArrayView<Value, Devices::Cuda> arr, 
{

    for (int i = threadIdx.x; i < arr.getSize(); i += blockDim.x)

    {

        const Value data = arr[i];

        const Value &data = arr[i];

        if (Cmp(data, pivot))

            smaller++;

        else if (Cmp(pivot, data))
@@ -85,6 +87,8 @@ __device__ void countElem(ArrayView<Value, Devices::Cuda> arr, 
    }

}



//-----------------------------------------------------------



template <typename Value, typename Function>

__device__ void copyDataShared(ArrayView<Value, Devices::Cuda> src,

                               ArrayView<Value, Devices::Cuda> dst,
@@ -98,7 +102,7 @@ __device__ void copyDataShared(ArrayView<Value, Devices::Cuda> src, 


    for (int i = threadIdx.x; i < src.getSize(); i += blockDim.x)

    {

        const Value data = src[i];

        const Value &data = src[i];

        if (Cmp(data, pivot))

            sharedMem[smallerOffset++] = data;

        else if (Cmp(pivot, data))
@@ -124,7 +128,7 @@ __device__ void copyData(ArrayView<Value, Devices::Cuda> src, 
{

    for (int i = threadIdx.x; i < src.getSize(); i += blockDim.x)

    {

        const Value data = src[i];

        const Value &data = src[i];

        if (Cmp(data, pivot))

        {

            /*

src/quicksort/quicksort.cuh

+111 −276

File changed.

Preview size limit exceeded, changes collapsed.

src/quicksort/quicksort_1Block.cuh

+69 −50

File changed.

Preview size limit exceeded, changes collapsed.

src/quicksort/quicksort_kernel.cuh

0 → 100644
+248 −0

File added.

Preview size limit exceeded, changes collapsed.