elvin-n commented on code in PR #11357:
URL: https://github.com/apache/tvm/pull/11357#discussion_r905980571
##########
src/driver/driver_api.cc:
##########
@@ -82,9 +82,12 @@ Target DefaultTargetHost(Target target) {
}
tir::Buffer BufferWithOffsetAlignment(Array<PrimExpr> shape, DataType dtype,
std::string name,
- int data_alignment, int offset_factor,
bool compact) {
+ int data_alignment, int offset_factor,
bool compact,
+ std::string memory_scope) {
DataType storage_dtype = (dtype == DataType::Bool() ? DataType::Int(8) :
dtype);
- auto data = tir::Var(name, PointerType(PrimType(storage_dtype)));
+ auto data =
+ tir::Var(name, memory_scope.empty() ?
PointerType(PrimType(storage_dtype))
+ :
PointerType(PrimType(storage_dtype), memory_scope));
Review Comment:
fixed in [PR11874](https://github.com/apache/tvm/pull/11874)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
