{config, pkgs, lib, ...}:
with lib;
let
  cfg = config.aviallon.hardware;
  nixos-nvidia-vgpu = import (builtins.fetchTarball "https://github.com/danielfullmer/nixos-nvidia-vgpu/archive/master.tar.gz") {
    inherit config;
    inherit pkgs;
    inherit lib;
  };
  useVgpu = (cfg.useProprietary &&
            (cfg.gpuVendor == "nvidia") &&
            (versionOlder config.boot.kernelPackages.kernel.version "5.10"));
in
{
  imports = [
    nixos-nvidia-vgpu
  ];

  config = mkIf useVgpu {
    hardware.nvidia.vgpu.enable = true; # Enable NVIDIA KVM vGPU + GRID driver
    hardware.nvidia.vgpu.unlock.enable = true; # Unlock vGPU functionality on consumer cards using DualCoder/vgpu_unlock project.
  };
}