compute_unit 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 82032a2c2110769430dd7b1a3f4b624227275eb5a3a3e62f6e0e42bdb41790c3
-  data.tar.gz: 7e45e2b1cd11b5cb0cfb40b00fee99a20ecc6e125064a2251372c4e14f22e1ca
+  metadata.gz: dcc45e864245505b64cd82233ca7f8ad70d0d12ca1a976862ba46c80e66fefc8
+  data.tar.gz: 6903aff2b613ee6f88c41aa051e65404f31c8923bab2059323dfd27bc99cc5e0
 SHA512:
-  metadata.gz: 8c818aa6ba28c5af8d57e8e99e4deb81a3eb4cc19532ce68d4888e6f644d780360c5568272667a9c5edc4807da90b49ee7c454c86164d925eab409c0332385c2
-  data.tar.gz: 2898757c00752709b8cfec4a4137f3bc65310d0e1af2a68637a1b3eba385f7e0188acb2790538b40e1fd2a650f65152175bfd99947cdd2d023baee3fa321aa64
+  metadata.gz: a971813c562bc8c29305f1fe3668f229b52bd018049d36879fcc943a935232e8d8bb34095e74643ca9d3da594796a4cdfdc33fe06b36a8501a75c0a330489cb3
+  data.tar.gz: bc17da953ab1904661c7037977c36b1dda69c7c4236c5854730743fa9332987a92ba3891e97052c4ee9361bea7a0b830a917235e15f1db75e8246b7835715a6e

data/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 ## Unrleased
 
+## 0.3.0
+Released 6/16/2020
+
+* Add name method to the cpu class
+* Use constant for proc path
+* Use debug when kernel file doesn't exist
+
 ## 0.2.1
 Released 6/6/2020
 

data/lib/compute_unit/cpu.rb

@@ -7,6 +7,8 @@ module ComputeUnit
     DEVICE_CLASS = '060000'
     DEVICE_CLASS_NAME = 'CPU'
     # @return [Array] - returns a list of device paths of all devices considered for display
+    alias name model
+
     def self.devices
       ComputeUnit::ComputeBase.devices.find_all do |device|
         ComputeUnit::Device.device_class(device) == DEVICE_CLASS

data/lib/compute_unit/device.rb

@@ -10,7 +10,7 @@ module ComputeUnit
   class Device
     # We can supply a mock sysfs path in order to test with containers and other scenarios
     SYSFS_DEVICES_PATH = File.join(ComputeUnit::SYSFS_PATH, 'bus', 'pci', 'devices')
-
+    PROC_PATH = ENV['PROC_PATH'] || '/proc'
     attr_reader :device_class_id,
                 :device_id,
                 :device_vendor_id,
@@ -130,7 +130,7 @@ module ComputeUnit
     rescue Errno::EINVAL, Errno::EPERM
       default
     rescue Errno::ENOENT
-      logger.warn("File #{path} does not exist, using defaults")
+      logger.debug("File #{path} does not exist, using defaults")
       default
     rescue Errno::EACCES
       logger.fatal('run this command as root or with sudo, using default value')
@@ -271,7 +271,7 @@ module ComputeUnit
                 logger.fatal(e.message)
                 default
               rescue Errno::ENOENT
-                logger.warn("File #{path} does not exist")
+                logger.debug("File #{path} does not exist")
                 default
               rescue Errno::EACCES
                 logger.fatal('Run this command as root or with sudo')

data/lib/compute_unit/gpus/nvidia_gpu.rb

@@ -8,6 +8,7 @@ module ComputeUnit
     MAKE = 'Nvidia'
     SUBTYPE = 'nvidia'
     NVIDIA_SMI = '/usr/bin/nvidia-smi'
+    NVIDIA_PROC_PATH = ENV['NVIDIA_PROC_PATH'] || File.join(ComputeUnit::Device::PROC_PATH, 'driver', 'nvidia', 'gpus')
 
     def initialize(device_path, opts = {})
       data = self.class.read_information_file(device_path).merge(opts)
@@ -159,7 +160,7 @@ module ComputeUnit
     def information_file
       @information_file ||= begin
         device_name = File.basename(device_path)
-        File.join('/proc/driver/nvidia/gpus', device_name, 'information')
+        File.join(NVIDIA_PROC_PATH, device_name, 'information')
       end
     end
 
@@ -209,7 +210,8 @@ module ComputeUnit
     #  :device_minor=>"7"}
     def self.read_information_file(device_path)
       device_name = File.basename(device_path)
-      information_file = "/proc/driver/nvidia/gpus/#{device_name}/information"
+      information_file = File.join(NVIDIA_PROC_PATH, device_name, 'information')
+
       File.open(information_file, 'r') do |file|
         content = file.read
         content.scan(/\n?([\w\s]*):\s+(.*)/).map { |key, value| [key.downcase.tr(' ', '_').to_sym, value] }.to_h

data/lib/compute_unit/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ComputeUnit
-  VERSION = '0.2.1'
+  VERSION = '0.3.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: compute_unit
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Corey Osman
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-06-07 00:00:00.000000000 Z
+date: 2020-06-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: opencl_ruby_ffi