---
id: pr-flashinfer-880
repo: flashinfer-ai/flashinfer
pr: 880
title: JIT compilation support for TVM
author: MasterJH5574
date: '2025-02-19'
url: https://github.com/flashinfer-ai/flashinfer/pull/880
source_category: upstream-code
architectures:
- sm100
tags: []
techniques: []
hardware_features: []
kernel_types: []
languages: []
captured_at: '2026-05-16'
status: merged
merge_sha: df05064de9172ee085d5f2cfd137d9621fbd0fa3
inclusion_reason: defer
changed_paths:
- CMakeLists.txt
- cmake/config.cmake
- custom_backend.py
- flashinfer/jit/__init__.py
- flashinfer/jit/attention/__init__.py
- flashinfer/jit/attention/pytorch.py
- flashinfer/jit/attention/tvm.py
- flashinfer/jit/attention/utils.py
- flashinfer/jit/env.py
- src/tvm_wrapper.cu
- tvm_binding/batch_decode.cu
- tvm_binding/batch_decode_customize_config.jinja
- tvm_binding/batch_decode_jit_tvm_binding.cu
- tvm_binding/batch_mla_config.jinja
- tvm_binding/batch_mla_jit_tvm_binding.cu
- tvm_binding/batch_mla_plan.cu
- tvm_binding/batch_mla_run.cu
- tvm_binding/batch_prefill.cu
- tvm_binding/batch_prefill_customize_config.jinja
- tvm_binding/batch_prefill_jit_tvm_binding.cu
- tvm_binding/batch_prefill_sm90.cu
- tvm_binding/batch_prefill_sm90_customize_config.jinja
- tvm_binding/batch_prefill_sm90_jit_tvm_binding.cu
- tvm_binding/tvm_binding_utils.h
artifact_dir: evidence/pull-bundles/flashinfer/gh-880
head_ref_oid: 2c27b99467f85e06ab98a283c79e661a8947e335
---
# JIT compilation support for TVM

- Repository: `flashinfer-ai/flashinfer`
- PR: [880](https://github.com/flashinfer-ai/flashinfer/pull/880)
- Decision: `defer`
- Reason: defer

## Changed Paths

- `CMakeLists.txt`
- `cmake/config.cmake`
- `custom_backend.py`
- `flashinfer/jit/__init__.py`
- `flashinfer/jit/attention/__init__.py`
- `flashinfer/jit/attention/pytorch.py`
- `flashinfer/jit/attention/tvm.py`
- `flashinfer/jit/attention/utils.py`
- `flashinfer/jit/env.py`
- `src/tvm_wrapper.cu`
- `tvm_binding/batch_decode.cu`
- `tvm_binding/batch_decode_customize_config.jinja`
- `tvm_binding/batch_decode_jit_tvm_binding.cu`
- `tvm_binding/batch_mla_config.jinja`
- `tvm_binding/batch_mla_jit_tvm_binding.cu`
- `tvm_binding/batch_mla_plan.cu`
- `tvm_binding/batch_mla_run.cu`
- `tvm_binding/batch_prefill.cu`
- `tvm_binding/batch_prefill_customize_config.jinja`
- `tvm_binding/batch_prefill_jit_tvm_binding.cu`
- `tvm_binding/batch_prefill_sm90.cu`
- `tvm_binding/batch_prefill_sm90_customize_config.jinja`
- `tvm_binding/batch_prefill_sm90_jit_tvm_binding.cu`
- `tvm_binding/tvm_binding_utils.h`

## Evidence Bundle

`evidence/pull-bundles/flashinfer/gh-880`
