forked from mindspore-Ecosystem/mindspore
clean code
This commit is contained in:
parent
a2f0a98814
commit
f2a5bdf26e
|
@ -22,7 +22,7 @@
|
|||
|
||||
namespace mindspore {
|
||||
namespace {
|
||||
std::string MemInfo2String(const std::string &label, const AddressPtrList &info) {
|
||||
std::string MemInfo2String(const std::string &label, const kernel::AddressPtrList &info) {
|
||||
std::ostringstream ss;
|
||||
ss << label << " " << info.size() << std::endl;
|
||||
for (size_t i = 0; i < info.size(); i++) {
|
||||
|
|
|
@ -26,16 +26,8 @@
|
|||
|
||||
namespace mindspore {
|
||||
namespace kernel {
|
||||
class Address;
|
||||
struct KernelLaunchInfo;
|
||||
using AddressPtr = std::shared_ptr<Address>;
|
||||
} // namespace kernel
|
||||
using AddressPtrList = std::vector<kernel::AddressPtr>;
|
||||
struct MemInfo {
|
||||
AddressPtrList *inputs_;
|
||||
AddressPtrList *workspaces_;
|
||||
AddressPtrList *outputs_;
|
||||
};
|
||||
class MemAddressRecorder : public BaseRecorder {
|
||||
public:
|
||||
MemAddressRecorder() {}
|
||||
|
|
Loading…
Reference in New Issue