[Relax] Allow ingesting tensor.chunk() from exported torch program (
#… Pull request merge
[Relax] Enable bfloat16 for softmax struct-info inference (
#17781 )
[Relax] Enable bfloat16 for softmax struct-info inference (
#17781 )
Pull request merge
yongwwwpushed 1 commit to main • 26041f8…90391bb • 20 hours ago
[Relax][Frontend] Support max/min in frontend op interface (
#17782 )
Pull request merge
tqchenpushed 1 commit to main • e60fd80…26041f8 • yesterday
[3rdparty] Enable bfloat16 for custom allreduce kernel (
#17780 )
Pull request merge
tqchenpushed 1 commit to main • 41c9c3b…e60fd80 • yesterday
[REFACTOR][TIR] remove legacy tir::any (
#17783 )
Pull request merge
yongwwwpushed 1 commit to main • c962198…41c9c3b • yesterday
[REFACTOR] Phase out StackVM (
#17784 )
Pull request merge
tqchenpushed 1 commit to main • 3da0738…c962198 • yesterday
[Relax][PyTorch] Support log2, log10 and log1p ops for ExportedProgra…
[Relax][PyTorch] Support log2, log10 and log1p ops for ExportedProgra…
Pull request merge
[Relax] Batch norm correctness on eval mode (
#17752 )
Pull request merge
[Relax] check for tensor_meta in exported_program_translator (
#17774 )
[Relax] check for tensor_meta in exported_program_translator (
#17774 )
Pull request merge
[Relax] Tensor.split with uneven tensors (
#17757 )
Pull request merge
[Relax][PyTorch] Add support for prod, std and var ops (
#17772 )
[Relax][PyTorch] Add support for prod, std and var ops (
#17772 )
Pull request merge
Hzfengsypushed 1 commit to main • e74ded2…adce5a4 • 5 days ago
[Relax][PyTorch] Add support for log2, log10 and log1p ops (
#17766 )
[Relax][PyTorch] Add support for log2, log10 and log1p ops (
#17766 )
Pull request merge
Hzfengsypushed 1 commit to main • a0d0859…e74ded2 • 7 days ago
[FIX][RELAX] fix fusion of transpose + matmul when constant weight (
#…
[FIX][RELAX] fix fusion of transpose + matmul when constant weight (
#… Pull request merge
Hzfengsypushed 1 commit to main • b0f1433…a0d0859 • 8 days ago
[Fix] Fix OpenCL header in attention utils (
#17762 )
[Fix] Fix OpenCL header in attention utils (
#17762 )
Pull request merge
[Relax][PyTorch] Add support for lerp, select and clone ops (
#17760 )
[Relax][PyTorch] Add support for lerp, select and clone ops (
#17760 )
Pull request merge
[Dlight] Fix general reduction rule to support non-last reduction axis (
Pull request merge
[Relax] Move TIR backend to gpu_generic (
#17749 )
[Relax] Move TIR backend to gpu_generic (
#17749 )
Pull request merge
tqchenpushed 1 commit to main • dc043fe…046c1ba • 11 days ago
[Relax][PyTorch] Support one_hot, empty_like ops for ExportedProgram …
Pull request merge
Hzfengsypushed 1 commit to main • f6236ce…dc043fe • 11 days ago
BugFix: Relax comment (
#17750 )
BugFix: Relax comment (
#17750 )
Pull request merge
tqchenpushed 1 commit to main • 435f641…f6236ce • 12 days ago
[Codegen] Support codegen for vectorized tir.ShuffleNode (
#17748 )
Pull request merge
tqchenpushed 1 commit to main • effdbef…435f641 • 12 days ago
You can’t perform that action at this time.