diff --git a/vgpu_unlock b/vgpu_unlock index 2549e04..9a8aab5 100755 --- a/vgpu_unlock +++ b/vgpu_unlock @@ -6,7 +6,7 @@ # This file is part of the "vgpu_unlock" project, and is distributed under the MIT License. # See the LICENSE file for more details. # -# Contributions from the vGPU Unlocking community included :) +# Contributions from Krutav Shah and the vGPU Unlocking community included :) # import errno @@ -105,8 +105,15 @@ script_source = r""" spoofed_devid = 0x13f2; // Tesla M60 for testing purposes } + // GM107 + if(actual_devid == 0x139a || // GTX 950M + actual_devid == 0x13b6) { // Quadro M1200 + spoofed_devid = 0x13f2; // Tesla M60 + } + + // GM204 - if(actual_devid == 0x13c3 || // GTX 960 OEM + if(actual_devid == 0x13c3 || // GTX 960 GM204 OEM Edition actual_devid == 0x13d9 || // GTX 965M actual_devid == 0x13d8 || // GTX 970M actual_devid == 0x13c2 || // GTX 970 @@ -148,7 +155,7 @@ script_source = r""" // GV100 (For the one person who owns a Titan Volta) if(actual_devid == 0x1d81 || // TITAN V - actual_devid == 0x1db6) { // TITAN V CEO Edition 32GB + actual_devid == 0x1db6)|| // TITAN V CEO Edition 32GB actual_devid == 0x1dbA) { // Quadro GV100 spoofed_devid = 0x1db4; // Tesla V100 } diff --git a/vgpu_unlock_hooks.c b/vgpu_unlock_hooks.c index 0dacd31..7efd44c 100644 --- a/vgpu_unlock_hooks.c +++ b/vgpu_unlock_hooks.c @@ -9,6 +9,9 @@ * Copyright 2021 Jonathan Johansson * This file is part of the "vgpu_unlock" project, and is distributed under the * MIT License. See the LICENSE file for more details. + * + * Contributions from Krutav Shah and the vGPU Unlocking community included :) + * */ /*------------------------------------------------------------------------------ @@ -561,7 +564,7 @@ static void vgpu_unlock_hmac_sha256(void* dst, */ /* Debug logs can be enabled here by changing 0 to 1. */ -#if 0 +#if 1 #define LOG(...) printk(__VA_ARGS__) #else #define LOG(...) @@ -766,6 +769,22 @@ static uint16_t vgpu_unlock_pci_devid_to_vgpu_capable(uint16_t pci_devid) { switch (pci_devid) { + /* GK104 Uses M60 profiles */ + case 0x1183: /* GTX 660 Ti */ + + /* GM107 Uses M60 profiles */ + case 0x139a: /* GTX 950M */ + case 0x13b6: /* Quadro M1200, GM107 */ + + /* GM204 */ + case 0x13c3: /* GTX 960 GM204 OEM Edition */ + case 0x13c2: /* GTX 970 */ + case 0x13c1: /* GM204 Unknown */ + case 0x13c0: /* GTX 980 */ + case 0x13f1: /* Quadro M4000 */ + case 0x13f0: /* Quadro M5000 */ + return 0x13f2; /* Tesla M60 */ + /* GP102 */ case 0x1b00: /* TITAN X (Pascal) */ case 0x1b02: /* TITAN Xp */ @@ -773,6 +792,16 @@ static uint16_t vgpu_unlock_pci_devid_to_vgpu_capable(uint16_t pci_devid) case 0x1b30: /* Quadro P6000 */ return 0x1b38; /* Tesla P40 */ + /* GP106 Uses P4 profiles*/ + case 0x1c09: /* P106-90 3GB */ + case 0x1c07: /* P106-100 6GB */ + case 0x1c04: /* GTX 1060 5GB */ + case 0x1c03: /* GTX 1060 6GB */ + case 0x1c02: /* GTX 1060 3GB */ + case 0x1c30: /* Quadro P2000 */ + case 0x1c31: /* Quadro P2200 */ + case 0x1C20: /* NVIDIA GeForce GTX 1060 with Max-Q Design (6GB variant) */ + /* GP104 */ case 0x1b80: /* GTX 1080 */ case 0x1b81: /* GTX 1070 */ @@ -782,6 +811,10 @@ static uint16_t vgpu_unlock_pci_devid_to_vgpu_capable(uint16_t pci_devid) case 0x1bb0: /* Quadro P5000 */ return 0x1bb3; /* Tesla P4 */ + /* GV100 */ + case 0x1d81: /* Titan V 16GB */ + return 0x1db4; /* Tesla V100 16GB PCIE */ + /* TU102 */ case 0x1e02: /* TITAN RTX */ case 0x1e04: /* RTX 2080 Ti */