Skip to content

TE Integration fails with TE main #2060

Closed
Closed
@kshitij12345

Description

@kshitij12345

Running the tests with latest TE main, pytest thunder/tests/test_transformer_engine_executor.py results in

      t26 = Tensor.to(x, torch.bfloat16, copy=True)  # t26: "cuda:0 bf16[32, 32]"
      t28 = Tensor.to(w, torch.bfloat16, copy=True)  # t28: "cuda:0 bf16[32, 32]"
>     (t6, (t12, t13, t14, t15, t16, t17, t18, t19), ctx_te_2311) = te_linear_22(t26, t28, None)
E     ValueError: not enough values to unpack (expected 8, got 6)

thunder.computation_37:17: ValueError

NOTE: This can't be reproduced on pjnl-latest as it seems to be pinned to TE v2.3

Metadata

Metadata

Assignees

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions