Skip to content

Commit

Permalink
Hiding implementation details for distance primitives (dense + sparse) (
Browse files Browse the repository at this point in the history
#344)

Authors:
  - Corey J. Nolet (https://github.com/cjnolet)

Approvers:
  - Divye Gala (https://github.com/divyegala)
  - Dante Gama Dessavre (https://github.com/dantegd)

URL: #344
  • Loading branch information
cjnolet authored Oct 22, 2021
1 parent a452192 commit ccba252
Show file tree
Hide file tree
Showing 40 changed files with 740 additions and 371 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,11 @@
*/

#pragma once
#include <raft/distance/pairwise_distance_base.cuh>
#include <raft/distance/detail/pairwise_distance_base.cuh>

namespace raft {
namespace distance {
namespace detail {

/**
* @brief the canberra distance matrix calculation implementer
Expand Down Expand Up @@ -157,5 +158,7 @@ void canberraImpl(int m, int n, int k, const InType *pA, const InType *pB,
n, m, k, lda, ldb, ldd, pB, pA, pDcast, fin_op, stream);
}
}

} // namespace detail
} // namespace distance
} // namespace raft
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,11 @@
*/

#pragma once
#include <raft/distance/pairwise_distance_base.cuh>
#include <raft/distance/detail/pairwise_distance_base.cuh>

namespace raft {
namespace distance {

namespace detail {
/**
* @brief the Chebyshev distance matrix calculation implementer
* It computes the following equation: cij = max(cij, op(ai-bj))
Expand Down Expand Up @@ -154,5 +154,6 @@ void chebyshevImpl(int m, int n, int k, const InType *pA, const InType *pB,
n, m, k, lda, ldb, ldd, pB, pA, pDcast, fin_op, stream);
}
}
} // namespace detail
} // namespace distance
} // namespace raft
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,12 @@

#pragma once
#include <raft/cuda_utils.cuh>
#include <raft/distance/pairwise_distance_base.cuh>
#include <raft/distance/detail/pairwise_distance_base.cuh>
#include <raft/linalg/reduce.cuh>

namespace raft {
namespace distance {
namespace detail {

/**
* @brief the Correlation distance matrix:
Expand Down Expand Up @@ -243,5 +244,7 @@ void correlationImpl(int m, int n, int k, const InType *pA, const InType *pB,
fin_op, stream);
}
}

} // namespace detail
} // namespace distance
} // namespace raft
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,12 @@

#pragma once

#include <raft/distance/pairwise_distance_base.cuh>
#include <raft/distance/detail/pairwise_distance_base.cuh>
#include <raft/linalg/norm.cuh>

namespace raft {
namespace distance {
namespace detail {

/**
* @brief the cosine distance matrix calculation implementer
Expand Down Expand Up @@ -201,5 +202,6 @@ void cosineAlgo1(Index_ m, Index_ n, Index_ k, const InType *pA,
}
}

}; // end namespace detail
}; // end namespace distance
}; // end namespace raft
Loading

0 comments on commit ccba252

Please sign in to comment.