Skip to content

Conversation

@pengcheng888
Copy link
Collaborator

@pengcheng888 pengcheng888 commented Jan 7, 2026

目标版本
main

功能描述
(1) 添加 infinicore.tensor函数,支持通过列表,numpy, 元组 ,scale 创建infinicore.Tensor
(2) 移除from_numpy函数的dtype参数

测试结果

测试新增的infinicore.tensor函数
Screenshot from 2026-01-07 17-57-40

测试修改的from_numpy函数
Screenshot from 2026-01-07 17-57-10

修改后,InfiniLM的llama.py和jiuge.py测试通过

Screenshot from 2026-01-07 18-02-42 Screenshot from 2026-01-07 18-03-44

@pengcheng888 pengcheng888 marked this pull request as ready for review January 7, 2026 10:06
@pengcheng888 pengcheng888 requested a review from a team January 7, 2026 10:06
@pengcheng888 pengcheng888 changed the title issue/892-添加infini.tensor函数; 移除from_numpy函数的dtype参数 issue/892-添加infinicore.tensor函数; 移除from_numpy函数的dtype参数 Jan 7, 2026
import torch

try:
import torch
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

后续在无torch的环境中运行 InfiniLM,所以需要额外处理 import torch

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[DEV] 添加infini.tensor函数; 移除from_numpy函数的dtype参数

2 participants