initial cuda build

[ROCm/composable_kernel commit: 2f2cf35bf4]
This commit is contained in:
Chao Liu
2018-10-22 11:51:10 -05:00
parent 6521ccba67
commit 6a5c465ad9
7 changed files with 179 additions and 82 deletions

View File

@@ -28,6 +28,8 @@ void TensorDescriptor::CalculateStrides()
mLens.rbegin(), mLens.rend() - 1, mStrides.rbegin() + 1, std::multiplies<std::size_t>());
}
DataType_t TensorDescriptor::GetDataType() const { return mDataType; }
std::size_t TensorDescriptor::GetDimension() const { return mLens.size(); }
std::size_t TensorDescriptor::GetElementSize() const