Program Listing for File NPUDeviceProp.h

↰ Return to documentation for file (csrc/backend/NPUDeviceProp.h)

#pragma once

#include <string>

namespace c10::backend {

struct NPUDeviceProp {
  std::string name{};
  size_t totalGlobalMem = 0;
};

} // namespace c10::npu