tvm
tvm copied to clipboard
apache
Open deep learning compiler stack for cpu, gpu and specialized accelerators
This PR is trying to fix issues https://github.com/apache/tvm/issues/18362. For now, return zeros with the correct shape since random initialization [randn](https://docs.pytorch.org/docs/stable/generated/torch.randn.html#torch-randn) is mainly used during training, not inference. This is a...
### Expected behavior A tiny Transformer-like block exported via torch.export crashes TVM when importing with tvm.relax.frontend.torch.from_exported_program(ep). Before the crash, PyTorch emits warnings that torch.export inserted a get_attr node without a...
The provided TIR module can be successfully compiled using `tvm.build` with target='llvm'. However, when using `ms.tir_integration.tune_tir` for Meta-Schedule tuning, the script crashes unexpectedly with a `tvm.tir.schedule.ScheduleError`. It seems a bug...
Bumps [form-data](https://github.com/form-data/form-data) from 3.0.1 to 3.0.4. Release notes Sourced from form-data's releases. v3.0.2 Fixes npmignore temporary build files (#532) move util.isArray to Array.isArray (#564) Tests migrate from travis to GHA...
![dependabot[bot] avatar](https://avatars.githubusercontent.com/in/29110?v=4 "dependabot[bot] avatar"){kind=avatar}
Simple reproducer is attached below. For detailed background, see https://github.com/tile-ai/tilelang/issues/1079 ### Expected behavior test program exit without error ### Actual behavior ``` $ TVM_BACKTRACE=1 python test.py error: Unexpected type for...
### Summary Importing a torch.exported STFT + RNN toy model fails in the TVM Relax Torch frontend with: ``` AssertionError: Unsupported function types ['rnn_tanh.input', 'real.default', 'unfold.default', 'imag.default', 'fft_fft.default'] ``` This...
### Summary Importing a torch.exported program into TVM Relax triggers a segmentation fault inside FFI during construction of a Relax Tuple. The minimal model performs a 4D advanced indexing write...
### Actual behavior Importing an exported PyTorch program fails early in TVM Relax Torch frontend with: ``` AssertionError: Unsupported function types ['randn.default'] ``` The model is intentionally small and uses...
Added logsigmoid operator support in the TVM Relax exported program and test script code to validate the relax module.
Work in progress, adds sequence padding to KVCache BeginForward to resolve a sequence padding issue with B200 GPUs and DeepSeek V3
