This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 61f6e7f428 [KVCache] PagedKVCache refactor, FlashInfer JIT and MLA
integration (#17674)
add 7bedfeb209 [Codegen] FP4 support (#17630)
No new revisions were added by this update.
Summary of changes:
ci/jenkins/docker-images.ini | 4 +-
ci/jenkins/unity_jenkinsfile.groovy | 4 +-
include/tvm/runtime/data_type.h | 22 ++++-
include/tvm/script/ir_builder/tir/ir.h | 2 +
include/tvm/tir/op.h | 2 +-
python/tvm/_ffi/runtime_ctypes.py | 7 ++
python/tvm/contrib/nvcc.py | 16 ++++
python/tvm/runtime/ndarray.py | 25 +++++-
python/tvm/script/ir_builder/tir/ir.py | 14 ++++
src/ir/expr.cc | 7 +-
src/runtime/ndarray.cc | 3 +
src/script/ir_builder/tir/ir.cc | 3 +
src/support/scalars.h | 3 +
src/target/llvm/codegen_llvm.cc | 2 +
src/target/source/codegen_c.cc | 2 +-
src/target/source/codegen_cuda.cc | 48 ++++++++++-
src/target/source/codegen_cuda.h | 6 +-
src/tir/op/op.cc | 10 +++
src/tir/transforms/dtype_conversion.cc | 2 +-
src/tir/transforms/dtype_conversion.h | 8 +-
.../python/codegen/test_target_codegen_cuda_fp4.py | 96 ++++++++++++++++++++++
21 files changed, 267 insertions(+), 19 deletions(-)
create mode 100644 tests/python/codegen/test_target_codegen_cuda_fp4.py
