笔试时间:2026年3月18日往年笔试合集:2023春招秋招笔试合集2024春招秋招笔试合集第一题:大模型训练显存优化算法题目在大模型训练中,为解决NPU显存不够用的情况,通常采用张量swap或者张量重计算的方式进行优化;张量swap是把张量的数据先搬到CPU,需要的时候,再从CPU搬回NPU;张量重计算是在需要张量的值时,在NPU上重新把张量的值计算出来。假设把模型部署到NPU上,还需要 m 大小的存储空间,才能让大模型运行起来;目前有 n 个候选张量,每个张量占用一定的存储空间;n 个张量中的每个张量都可以进行swap或者重计算,swap和重计算分别对应不同的代价;试着编写一段程序,从 n...