[PATCH 2/3] Provide nvidia.ko symlinks to nvidia-assemble snap common directory

[Dimitri John Ledkov]

Signed-off-by: Dimitri John Ledkov <dimitri.ledkov at canonical.com>
---
 snapcraft.yaml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/snapcraft.yaml b/snapcraft.yaml
index 5391553aef..47e87a2c0e 100644
--- a/snapcraft.yaml
+++ b/snapcraft.yaml
@@ -52,6 +52,12 @@ parts:
         "$SNAPCRAFT_PART_SRC"/trim-firmware "$SNAPCRAFT_PART_INSTALL"/lib
         # Clean nvidia modules
         sh CLEAN
+        # Fake the nvidia.ko modules pointing at an nvidia assemble location
+        cd ../
+        for module in nvidia-drm.ko nvidia-modeset.ko nvidia-peermem.ko nvidia-uvm.ko nvidia.ko
+        do
+          ln -s /var/snap/nvidia-assemble/common/nvidia-driver/$module $module
+        done
       else
         # Update depmod
         depmod -b "$SNAPCRAFT_PART_INSTALL" $(basename "$SNAPCRAFT_PART_INSTALL"/lib/modules/*)
-- 
2.34.1