Skip to content

Commit

Permalink
Address comments
Browse files Browse the repository at this point in the history
  • Loading branch information
weberlo committed Nov 24, 2019
1 parent 013795e commit 9d77f82
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 12 deletions.
4 changes: 2 additions & 2 deletions python/tvm/micro/base.py
Expand Up @@ -138,11 +138,11 @@ def _check_system(self):
Raises error if not supported.
"""
if not sys.platform.startswith("linux"):
raise RuntimeError("MicroTVM is currently only supported on Linux")
raise RuntimeError("MicroTVM is currently only supported on Linux hosts")
# TODO(weberlo): Add 32-bit support.
# It's primarily the compilation pipeline that isn't compatible.
if sys.maxsize <= 2**32:
raise RuntimeError("MicroTVM is currently only supported on 64-bit platforms")
raise RuntimeError("MicroTVM is currently only supported on 64-bit host platforms")

def __enter__(self):
self._enter()
Expand Down
13 changes: 3 additions & 10 deletions src/runtime/micro/micro_session.h
Expand Up @@ -302,11 +302,7 @@ struct TVMArray32 {
byte_offset(byte_offset.val32),
pad2(0) { }

/*!
* \brief The opaque data pointer points to the allocated data.
* This will be CUDA device pointer or cl_mem handle in OpenCL.
* This pointer is always aligns to 256 bytes as in CUDA.
*/
/*! \brief opaque pointer to the allocated data */
uint32_t data;
/*! \brief The device context of the tensor */
DLContext ctx;
Expand Down Expand Up @@ -349,11 +345,8 @@ struct TVMArray64 {
shape(shape.val64),
strides(strides.val64),
byte_offset(byte_offset.val64) { }
/*!
* \brief The opaque data pointer points to the allocated data.
* This will be CUDA device pointer or cl_mem handle in OpenCL.
* This pointer is always aligns to 256 bytes as in CUDA.
*/

/*! \brief opaque pointer to the allocated data */
uint64_t data;
/*! \brief The device context of the tensor */
DLContext ctx;
Expand Down

0 comments on commit 9d77f82

Please sign in to comment.