Implement a few additional services

Author: adurang

offload/plugins-nextgen/level_zero/include/L0Device.h

@@ -351,6 +351,23 @@ class L0DeviceTy final : public GenericDeviceTy {
   uint32_t getMaxGroupSize() const {
     return ComputeProperties.maxTotalGroupSize;
   }
+  uint32_t getMaxGroupCount() const {
+    return getMaxGroupCountX() * getMaxGroupCountY() * getMaxGroupCountZ();
+  }
+
+  uint32_t getMaxGroupSizeX() const { return ComputeProperties.maxGroupSizeX; }
+  uint32_t getMaxGroupSizeY() const { return ComputeProperties.maxGroupSizeY; }
+  uint32_t getMaxGroupSizeZ() const { return ComputeProperties.maxGroupSizeZ; }
+  uint32_t getMaxGroupCountX() const {
+    return ComputeProperties.maxGroupCountX;
+  }
+  uint32_t getMaxGroupCountY() const {
+    return ComputeProperties.maxGroupCountY;
+  }
+  uint32_t getMaxGroupCountZ() const {
+    return ComputeProperties.maxGroupCountZ;
+  }
+  uint32_t getMemoryClockRate() const { return MemoryProperties.maxClockRate; }
   uint64_t getGlobalMemorySize() const { return MemoryProperties.totalSize; }
   size_t getCacheSize() const { return CacheProperties.cacheSize; }
   uint64_t getMaxMemAllocSize() const {
@@ -597,6 +614,12 @@ class L0DeviceTy final : public GenericDeviceTy {
   }
 
   Expected<InfoTreeNode> obtainInfoImpl() override;
+  uint64_t getClockFrequency() const override { return DeviceProperties.coreClockRate; }
+  uint64_t getHardwareParallelism() const override { return getTotalThreads(); }
+  Error getDeviceMemorySize(uint64_t &DSize) {
+    DSize = getGlobalMemorySize();
+    return Plugin::success();
+  }
 
   Error getDeviceStackSize(uint64_t &V) override {
     V = 0;

offload/plugins-nextgen/level_zero/src/L0Device.cpp

@@ -563,6 +563,20 @@ Expected<InfoTreeNode> L0DeviceTy::obtainInfoImpl() {
   Info.add("Number of slices", getNumSlices());
   Info.add("Max Group size", getMaxGroupSize(), "",
            DeviceInfo::MAX_WORK_GROUP_SIZE);
+  auto &MaxGroupSize =
+      *Info.add("Workgroup Max Size per Dimension", std::monostate{}, "",
+                DeviceInfo::MAX_WORK_GROUP_SIZE_PER_DIMENSION);
+  MaxGroupSize.add("x", getMaxGroupSizeX());
+  MaxGroupSize.add("y", getMaxGroupSizeY());
+  MaxGroupSize.add("z", getMaxGroupSizeZ());
+  Info.add("Maximum Grid Dimensions", getMaxGroupSize() * getMaxGroupCount(),
+           "", DeviceInfo::MAX_WORK_SIZE);
+  auto &MaxSize = *Info.add("Grid Size per Dimension", std::monostate{}, "",
+                            DeviceInfo::MAX_WORK_SIZE_PER_DIMENSION);
+  MaxSize.add("x", getMaxGroupSizeX() * getMaxGroupCountX());
+  MaxSize.add("y", getMaxGroupSizeY() * getMaxGroupCountY());
+  MaxSize.add("z", getMaxGroupSizeZ() * getMaxGroupCountZ());
+
   Info.add("Local memory size (bytes)", getMaxSharedLocalMemory());
   Info.add("Global memory size (bytes)", getGlobalMemorySize(), "",
            DeviceInfo::GLOBAL_MEM_SIZE);
@@ -571,6 +585,10 @@ Expected<InfoTreeNode> L0DeviceTy::obtainInfoImpl() {
            DeviceInfo::MAX_MEM_ALLOC_SIZE);
   Info.add("Max clock frequency (MHz)", getClockRate(), "",
            DeviceInfo::MAX_CLOCK_FREQUENCY);
+  Info.add("Max memory clock frequency (MHz)", getMemoryClockRate(), "",
+           DeviceInfo::MEMORY_CLOCK_RATE);
+  Info.add("Memory Address Size", uint64_t{64u}, "bits",
+           DeviceInfo::ADDRESS_BITS);
   return Info;
 }