youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix issue/1138

Browse Source
This commit is contained in:
ionthruster
2023-10-24 12:02:15 -07:00
parent 5cd735c48e
commit 6581237a48
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/cute/pointer.hpp
View File

@ -93,8 +93,6 @@ struct device_ptr
{ {
using value_type = T; using value_type = T;
static const uint32_t ElementsPerStoredItem = sizeof(T) * 8 / sizeof_bits_v<T>;
CUTE_HOST_DEVICE constexpr CUTE_HOST_DEVICE constexpr
device_ptr(T* ptr) : ptr_(ptr) {} device_ptr(T* ptr) : ptr_(ptr) {}
@ -113,7 +111,7 @@ struct device_ptr
template <class Index> template <class Index>
CUTE_HOST_DEVICE constexpr CUTE_HOST_DEVICE constexpr
DerivedType operator+(Index const& i) const { return {ptr_ + i / ElementsPerStoredItem}; } DerivedType operator+(Index const& i) const { return {ptr_ + i}; }
CUTE_HOST_DEVICE constexpr friend CUTE_HOST_DEVICE constexpr friend
ptrdiff_t operator-(device_ptr<T,DerivedType> const& a, ptrdiff_t operator-(device_ptr<T,DerivedType> const& a,