[PATCH] D80118: Unrank mcuMemHostRegister tensor argument.

[Christian Sigg via Phabricator via llvm-commits]

csigg marked 2 inline comments as done.
csigg added inline comments.


================
Comment at: mlir/tools/mlir-cuda-runner/cuda-runtime-wrappers.cpp:106
+extern "C" void mcuMemHostRegisterFloat(int64_t rank, void *ptr) {
+  auto *desc = static_cast<StridedMemRefType<float, 1> *>(ptr);
+  auto sizes = llvm::ArrayRef<int64_t>(desc->sizes, rank);
----------------
herhut wrote:
> This is a bit fishy, as you rely on the inner layout of the descriptor declared in a different file. It would be nicer to have an interface in that header that allows to do this.
> 
> 
Will do that in a separate change.


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D80118/new/

https://reviews.llvm.org/D80118