detect_device.cc 453 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/*!
 *  Copyright (c) 2018 by Contributors
 * \file detect_device.cc
 */

#include <tvm/ir_pass.h>
#include <tvm/ir_mutator.h>
#include "../pass/ir_util.h"

namespace tvm {
namespace ir {
Stmt DecorateDeviceScope(Stmt stmt) {
  Stmt body = AttrStmt::make(make_zero(Int(32)),
                             ir::attr::device_scope,
                             0,
                             stmt);
  return body;
}

}  // namespace ir
}  // namespace tvm