Skip to content

Commit

Permalink
cuml-#1 - Removing utils.h from dbscan. Using cuda_utils.h supplied i…
Browse files Browse the repository at this point in the history
…n ml_prims
  • Loading branch information
Corey Nolet committed Nov 14, 2018
1 parent 7a02237 commit de30217
Show file tree
Hide file tree
Showing 9 changed files with 16 additions and 70 deletions.
2 changes: 1 addition & 1 deletion cuML/src/dbscan/epsneigh/algo1.h
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#pragma once

#include "cuda_utils.h"
#include <cuda_utils.h>
#include "pack.h"

namespace Dbscan {
Expand Down
2 changes: 1 addition & 1 deletion cuML/src/dbscan/epsneigh/algo2.h
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#pragma once

#include "../cuda_utils.h"
#include <cuda_utils.h>
#include "../types.h"
#include "pack.h"

Expand Down
2 changes: 1 addition & 1 deletion cuML/src/dbscan/epsneigh/algo3.h
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#pragma once

#include "../cuda_utils.h"
#include <cuda_utils.h>
#include "../types.h"
#include "pack.h"

Expand Down
2 changes: 1 addition & 1 deletion cuML/src/dbscan/epsneigh/algo4.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

#pragma once

#include "../cuda_utils.h"
#include <cuda_utils.h>
#include "pack.h"
#include "../common.h"

Expand Down
2 changes: 1 addition & 1 deletion cuML/src/dbscan/epsneigh/naive.h
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#pragma once

#include "../cuda_utils.h"
#include <cuda_utils.h>
#include "pack.h"

namespace Dbscan {
Expand Down
2 changes: 1 addition & 1 deletion cuML/src/dbscan/labelling/runner.h
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#pragma once

#include "utils.h"
#include <cuda_utils.h>
#include <iostream>
#include <limits>
#include "naive.h"
Expand Down
20 changes: 10 additions & 10 deletions cuML/src/dbscan/runner.h
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#pragma once

#include "utils.h"
#include <cuda_utils.h>
#include "vertexdeg/runner.h"
#include "adjgraph/runner.h"
#include "labelling/runner.h"
Expand Down Expand Up @@ -69,15 +69,15 @@ size_t run(Type_f* x, Type N, Type D, Type_f eps, Type minPts, Type* labels,
cudaStream_t stream) {
const size_t align = 256;
int batchSize = ceildiv(N, nBatches);
size_t adjSize = alignSize<size_t>(sizeof(bool) * N * batchSize, align);
size_t corePtsSize = alignSize<size_t>(sizeof(bool) * N, align);
size_t visitedSize = alignSize<size_t>(sizeof(bool) * N, align);
size_t xaSize = alignSize<size_t>(sizeof(bool) * N, align);
size_t mSize = alignSize<size_t>(sizeof(bool), align);
size_t vdSize = alignSize<size_t>(sizeof(Type) * (batchSize + 1), align);
size_t exScanSize = alignSize<size_t>(sizeof(Type) * batchSize, align);
size_t mapIdSize = alignSize<size_t>(sizeof(Type) * N, align);
size_t dotsSize = alignSize<size_t>(sizeof(Type_f) * N, align);
size_t adjSize = alignTo<size_t>(sizeof(bool) * N * batchSize, align);
size_t corePtsSize = alignTo<size_t>(sizeof(bool) * N, align);
size_t visitedSize = alignTo<size_t>(sizeof(bool) * N, align);
size_t xaSize = alignTo<size_t>(sizeof(bool) * N, align);
size_t mSize = alignTo<size_t>(sizeof(bool), align);
size_t vdSize = alignTo<size_t>(sizeof(Type) * (batchSize + 1), align);
size_t exScanSize = alignTo<size_t>(sizeof(Type) * batchSize, align);
size_t mapIdSize = alignTo<size_t>(sizeof(Type) * N, align);
size_t dotsSize = alignTo<size_t>(sizeof(Type_f) * N, align);
if(workspace == NULL) {
auto size = adjSize
+ corePtsSize
Expand Down
1 change: 0 additions & 1 deletion cuML/src/dbscan/vertexdeg/algo5.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
#pragma once

#include <cuda_utils.h>
#include "utils.h"
#include "pack.h"
#include "../common.h"
#include <cub/cub.cuh>
Expand Down
53 changes: 0 additions & 53 deletions cuML/src/utils.h

This file was deleted.

0 comments on commit de30217

Please sign in to comment.