Skip to content

Latest commit

 

History

History
34 lines (20 loc) · 1.18 KB

File metadata and controls

34 lines (20 loc) · 1.18 KB

姓名

许轩昂

实习项目

项目二:模型迁移工具建设

本周工作

  1. 新增torch.distributed.all_gather_object,torch.distributed.all_to_all_single,torch.distributed.reduce_scatter_tensor等API的映射规则
  1. 新增/修改torch.Tensor.is_contiguous,torch.cuda.reset_max_memory_cached,torch.cuda.reset_max_memory_allocated等API的映射规则。
  1. 修改torch.Tensor.bernoulli_,torch.Tensor.cholesky_inverse,torch.Tensor.ldexp,torch.Tensor.logaddexp等API的映射规则。
  1. 新增/修改torch.Tensor.baddbmm,torch.Tensor.baddbmm_,torch.Tensor.lu_solve等API的映射规则。

下周工作

  1. 完成剩余约20个需要开发的API映射关系的开发,包括映射文档,转换规则,Matcher以及单测。

导师点评