Skip to content

Conversation

@tianyuxbear
Copy link
Contributor

@tianyuxbear tianyuxbear commented Aug 11, 2025

完成T1-1-2赛题所有算子的实现代码、Pytorch单元测试代码以及GGUF测试代码。
算子设计文档详见设计文档PR:InfiniTensor/InfiniCore-Documentation#37

测试情况

T1-1-2赛题中所有Pytorch单元测试脚本均已放入scripts/python_test.py中。

NVIDIA平台

nvidia-test-1 nvidia-test-2

METAX平台

metax-test-1 metax-test-2

ILUVATAR平台

iluvatar-test-1 iluvatar-test-2

实现情况

silu

commit id: bc6b0ec

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。
  • 修改test/infiniop/libinfiniop/utils.py基建,以更新TestTensor(output)中的torch_tensor为计算得到的新tensor。
    image

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/silu

div

commit id: e832909

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/div

logical_and

commit id: 147b458

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。
  • 修改test/infiniop/libinfiniop/utils.py基建,以支持TestTensor中的BOOL以及INT类型。
    image

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/logical_and

logical_or

commit id: 9eb53ac

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/logical_or

equal(对齐torch.eq)

commit id: 3b72eab

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/equal

all_equal(对齐torch.equal)

commit id: e31492b

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/all_equal

relu_backward

commit id: 4985e45

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/relu_backward

gelu

commit id: 99f3794

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/gelu

gelu_backward

commit id: 9ea3cab

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/gelu_backward

cross_entropy_loss_backward

commit id: 7410148

  • 实现CPU和NVIDIA 平台的算子实现代码、Pytorch单元测试代码以及GGUF测试代码。

commit id: 6c2bda3

  • 实现METAX 平台的算子实现代码。

文档

https://github.com/tianyuxbear/InfiniCore-Documentation/tree/submission/infiniop/ops/cross_entropy_loss_backward

fix: modify some codes to profile
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant