gpu memory management code

Hi Guys,

I am looking at GPU memory management code in Nvidia Driver for PCI card GPU.
It is very clear to find the gpu memory management code in the PC driver such as uvm8_pmm_gpu.c
Can anyone let me know where or what file/folder in kernel I need to look into for the gpu memory management code for TX2 or Xavier?

I am using the kernel code in BSP package from the website below.

Thanks,

The gpu driver code is inside “/kernel/nvgpu/drivers/gpu” if you are using rel-32 based kernel.