Skip to content

Commit

Permalink
update #1 after review
Browse files Browse the repository at this point in the history
  • Loading branch information
megazone87 committed Aug 27, 2020
1 parent 0e93db9 commit 2bee88a
Show file tree
Hide file tree
Showing 6 changed files with 330 additions and 190 deletions.
2 changes: 1 addition & 1 deletion .clang-format
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ BasedOnStyle: Google
---
Language: Cpp
Cpp11BracedListStyle: true
Standard: Cpp11
Standard: c++11
DerivePointerAlignment: false
PointerAlignment: Right
---
12 changes: 10 additions & 2 deletions k2/csrc/cuda/arch.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,15 @@
namespace k2 {

/**
* K2_PTX_ARCH get the target PTX version through the active compiler (or zero for the host).
* @def K2_PTX_ARCH
*
* @brief
* K2_PTX_ARCH get the target PTX version through the active compiler.
*
* @details
* - for host, it's 0.
* - for device, it's `__CUDA_ARCH__`, which indicates the
* compute compatibility, should >= 200.
*/
#ifndef K2_PTX_ARCH
#ifndef __CUDA_ARCH__
Expand All @@ -19,6 +27,6 @@ namespace k2 {
#endif
#endif

} // end namespace k2
} // namespace k2

#endif // K2_CSRC_CUDA_ARCH_H_
2 changes: 1 addition & 1 deletion k2/csrc/cuda/context.cu
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include <cstdlib>

#include "k2/csrc/cuda/context.h"
#include "k2/csrc/cuda/debug.cuh"
#include "k2/csrc/cuda/debug.h"
#include "k2/csrc/cuda/error.h"

static constexpr size_t kAlignment = 64;
Expand Down
169 changes: 0 additions & 169 deletions k2/csrc/cuda/debug.cuh

This file was deleted.

Loading

0 comments on commit 2bee88a

Please sign in to comment.