123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216 |
- /*
- Copyright (c) 2017-2018, Feral Interactive
- All rights reserved.
- Redistribution and use in source and binary forms, with or without
- modification, are permitted provided that the following conditions are met:
- * Redistributions of source code must retain the above copyright notice,
- this list of conditions and the following disclaimer.
- * Redistributions in binary form must reproduce the above copyright
- notice, this list of conditions and the following disclaimer in the
- documentation and/or other materials provided with the distribution.
- * Neither the name of Feral Interactive nor the names of its contributors
- may be used to endorse or promote products derived from this software
- without specific prior written permission.
- THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
- AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
- LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
- CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
- SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
- INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
- CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
- POSSIBILITY OF SUCH DAMAGE.
- */
- #define _GNU_SOURCE
- #include "logging.h"
- #include "external-helper.h"
- #include "gpu-control.h"
- // TODO
- // Apply Nvidia GPU settings (CoolBits will be needed)
- // Apply AMD GPU settings (Will need user changing pwm1_enable)
- // Provide documentation on optimisations
- // Intel?
- // Gather GPU type and information automatically if possible
- /* Helper to quit with usage */
- static const char *usage_text =
- "usage: gpuclockctl PCI_ID DEVICE [get] [set CORE MEM [PERF_LEVEL]]]";
- static void print_usage_and_exit(void)
- {
- fprintf(stderr, "%s\n", usage_text);
- exit(EXIT_FAILURE);
- }
- /**
- * Get the gpu state
- * Populates the struct with the GPU info on the system
- */
- int get_gpu_state(struct GameModeGPUInfo *info)
- {
- fprintf(stderr, "Fetching GPU state is currently unimplemented!\n");
- return info != NULL;
- }
- /**
- * Set the gpu state based on input parameters on Nvidia
- */
- int set_gpu_state_nv(struct GameModeGPUInfo *info)
- {
- if (info->vendor != Vendor_NVIDIA)
- return -1;
- // These commands don't technically even need root
- /* Set the GPUGraphicsClockOffset parameter */
- char core_arg[64];
- snprintf(core_arg,
- 64,
- "[gpu:%ld]/GPUGraphicsClockOffset[%ld]=%ld",
- info->device,
- info->nv_perf_level,
- info->core);
- const char *exec_args_core[] = { "/usr/bin/nvidia-settings", "-a", core_arg, NULL };
- if (run_external_process(exec_args_core) != 0) {
- LOG_ERROR("ERROR: Failed to set %s!\n", core_arg);
- return -1;
- }
- /* Set the GPUMemoryTransferRateOffset parameter */
- char mem_arg[64];
- snprintf(mem_arg,
- 64,
- "[gpu:%ld]/GPUMemoryTransferRateOffset[%ld]=%ld",
- info->device,
- info->nv_perf_level,
- info->mem);
- const char *exec_args_mem[] = { "/usr/bin/nvidia-settings", "-a", mem_arg, NULL };
- if (run_external_process(exec_args_mem) != 0) {
- LOG_ERROR("ERROR: Failed to set %s!\n", mem_arg);
- return -1;
- }
- return 0;
- }
- /**
- * Set the gpu state based on input parameters on amd
- */
- int set_gpu_state_amd(struct GameModeGPUInfo *info)
- {
- if(info->vendor != Vendor_AMD)
- return -1;
- // We'll want to set both the following:
- // core: device/pp_sclk_od (0%+ additional)
- // mem: device/pp_mclk_od (0%+ additional)
- // Guide from https://www.maketecheasier.com/overclock-amd-gpu-linux/
- fprintf(stderr, "Setting GPU parameters on AMD is currently unimplemented!\n");
- return 0;
- }
- /* Helper to get and verify vendor value */
- static long get_vendor(const char *val)
- {
- char *end;
- long ret = strtol(val, &end, 0);
- if (!GPUVendorValid(ret) || end == val) {
- LOG_ERROR("ERROR: Invalid GPU Vendor passed (0x%04x)!\n", (unsigned short)ret);
- print_usage_and_exit();
- }
- return ret;
- }
- /* Helper to get and verify device value */
- static long get_device(const char *val)
- {
- char *end;
- long ret = strtol(val, &end, 10);
- if (ret < 0 || end == val) {
- LOG_ERROR("ERROR: Invalid GPU device passed (%ld)!\n", ret);
- print_usage_and_exit();
- }
- return ret;
- }
- /* Helper to get and verify core and mem value */
- static long get_generic_value(const char *val)
- {
- char *end;
- long ret = strtol(val, &end, 10);
- if (ret < 0 || end == val) {
- LOG_ERROR("ERROR: Invalid value passed (%ld)!\n", ret);
- print_usage_and_exit();
- }
- return ret;
- }
- /**
- * Main entry point, dispatch to the appropriate helper
- */
- int main(int argc, char *argv[])
- {
- if (argc == 4 && strncmp(argv[3], "get", 3) == 0) {
- /* Get and verify the vendor and device */
- struct GameModeGPUInfo info;
- memset(&info, 0, sizeof(info));
- info.vendor = get_vendor(argv[1]);
- info.device = get_device(argv[2]);
- /* Fetch the state and print it out */
- get_gpu_state(&info);
- printf("%ld %ld\n", info.core, info.mem);
- } else if (argc >= 6 && argc <= 7 && strncmp(argv[3], "set", 3) == 0) {
- /* Must be root to set the state */
- if (geteuid() != 0) {
- fprintf(stderr, "gpuclockctl must be run as root to set values\n");
- print_usage_and_exit();
- }
- /* Get and verify the vendor and device */
- struct GameModeGPUInfo info;
- memset(&info, 0, sizeof(info));
- info.vendor = get_vendor(argv[1]);
- info.device = get_device(argv[2]);
- info.core = get_generic_value(argv[4]);
- info.mem = get_generic_value(argv[5]);
- if (info.vendor == Vendor_NVIDIA)
- info.nv_perf_level = get_generic_value(argv[6]);
- printf("gpuclockctl setting core:%ld mem:%ld on device:%ld with vendor 0x%04x\n",
- info.core,
- info.mem,
- info.device,
- (unsigned short)info.vendor);
- if (info.vendor == Vendor_NVIDIA)
- printf("on Performance Level %ld\n", info.nv_perf_level);
- switch (info.vendor) {
- case Vendor_NVIDIA:
- return set_gpu_state_nv(&info);
- case Vendor_AMD:
- return set_gpu_state_amd(&info);
- default:
- printf("Currently unsupported GPU vendor 0x%04x, doing nothing!\n", (short)info.vendor);
- break;
- }
- } else {
- print_usage_and_exit();
- }
- return EXIT_SUCCESS;
- }
|