[PyTorch][torch.compile] Add TensorProto mechanism - #8

Open

pggPL wants to merge 46 commits into

mainfrom

tensor_proto_mechanism

Commits on Jul 7, 2026

Add TensorProto mechanism for data-free quantized tensor allocation

pggPL
and
claude
committed
6d3eece
[PyTorch] torch.compile: dedup cached FP8 weight from saved-for-backward

pggPL
and
claude
committed
4bdaa60
[PyTorch] nvfp4: emit _describe_buffers in canonical flatten order

pggPL
and
claude
committed
3be6fc7
Address review: error on undescribed buffers, gate nvfp4 test on HW support

pggPL
and
claude
committed
8a9d90c
[PyTorch] Workaround torch.compile staticmethod guard bug in NVFP4 _describe_buffers

pggPL
and
claude
committed
e36cf6d

Commits on Jul 13, 2026

[PyTorch] Document TensorOrQuantized union (review feedback)
pggPL
committed
90fc494
[PyTorch] Add shape property to QuantizedTensorStorage
pggPL
committed
a3bf5c1
[PyTorch] Test parity of Python alloc vs C++ make_empty
pggPL
committed
37b0d17
[PyTorch] Simplify comment on captured requires_grad flags
pggPL
committed
4feacdc
[PyTorch] Drop redundant None guards in wt_save alias dedup
pggPL
committed
5adb6e4
[PyTorch] Use torch._prims_common.make_contiguous_strides_for
pggPL
committed
40d2d7b
[PyTorch] Raise on update_usage of a non-quantized TensorProto
pggPL
committed
78e95a6
[PyTorch] Collapse to_tensor_proto branches
pggPL
committed
b3229cc
[PyTorch] Drop dead weight_fp8 fallback in fake backward
pggPL
committed
2b19368
[PyTorch] Drop redundant comment on saved_weight
pggPL
committed
c3dec83
Merge branch 'main' of https://github.com/NVIDIA/TransformerEngine into tensor_proto_mechanism
pggPL
committed
cb53cc3

Commits on Jul 15, 2026

[PyTorch] Add TensorProto.assemble for rebuilding from ready-made buffers
pggPL
committed
dd99206

Commits on Jul 27, 2026

Merge remote-tracking branch 'upstream/main' into tensor_proto_mechanism
pggPL
committed
2c28aa1

Commits on Jul 28, 2026

Commits on Jul 29, 2026

Commits on Jul 31, 2026

Commits on Aug 3, 2026

[CI] Pin JAX image to 2026-07-21 (#3250)

fheinecke
and
ksivaman
authored
a9e635a
[Common] Remove `nv-internal-*` comments (#3280)
ksivaman
authored
6b23ad3
Pin nltk to work around breaking change in tests (#3306)
fheinecke
authored
cb7e2de

Commits on Aug 4, 2026

[JAX] EP Dispatch with overflow detection option (#3277)
phu0ngng
authored
72f4e23
Rename TensorProto to TensorSpec and tighten the inner-tensor contract

pggPL
and
claude
committed
0fbb4af
Restore three unrelated comments mangled by the buffers/inner-tensors rename

pggPL
and
claude
committed
9de57b3
[JAX] Add NLTK pin <3.10.1 in encoder requirements (#3314)
aybchan
authored
82a0f6e
Fix typo in contribution guidelines (#3308)
cupkk
authored
429c1bc
[Pytorch][Common] Hybrid quantization (#2817)

authored
bce8162
Merge branch 'main' into tensor_proto_mechanism

pggPL
and
claude
committed
5612f13