681f85a8bc34e43f6f14aa952f9cb2e043099f57 - collateral

commit	681f85a8bc34e43f6f14aa952f9cb2e043099f57	[log] [tgz]
author	TimAtGoogle <tjharvey@google.com>	Tue Dec 10 19:11:07 2024 -0600
committer	TimAtGoogle <tjharvey@google.com>	Tue Dec 10 19:11:07 2024 -0600
tree	8d0c8edbbb935c2a7c2cf27b216b85eba9ae6b9e
parent	b71e0295fde88695343dce3d5a3c1e17ceb04c81 [diff]

google3/xla/mosaic

mosaic/BUILD[Added - diff]
mosaic/OWNERS[Added - diff]
mosaic/custom_call_kernel_name.cc[Added - diff]
mosaic/custom_call_kernel_name.h[Added - diff]
mosaic/dialect/llo/llo.td[Added - diff]
mosaic/dialect/llo/llo_builder.cc[Added - diff]
mosaic/dialect/llo/llo_builder.h[Added - diff]
mosaic/dialect/llo/llo_builder_gen.cc[Added - diff]
mosaic/dialect/llo/llo_dialect.cc[Added - diff]
mosaic/dialect/llo/llo_dialect.h[Added - diff]
mosaic/dialect/llo/llo_ops.cc[Added - diff]
mosaic/dialect/tpu/tpu_passes.h[Added - diff]
mosaic/dialect/tpu/tpu_passes.td[Added - diff]
mosaic/dialect/tpu/transforms/lower_to_llo.cc[Added - diff]
mosaic/examples/BUILD[Added - diff]
mosaic/examples/benchmark.py[Added - diff]
mosaic/examples/block_sparse_matmul.py[Added - diff]
mosaic/examples/build_defs.bzl[Added - diff]
mosaic/examples/cholesky.py[Added - diff]
mosaic/examples/collective_matmul.py[Added - diff]
mosaic/examples/collective_matmul_test.py[Added - diff]
mosaic/examples/flash_attention.py[Added - diff]
mosaic/examples/gather1d_sparsecore.py[Added - diff]
mosaic/examples/gather2d_sparsecore.py[Added - diff]
mosaic/examples/matmul.py[Added - diff]
mosaic/examples/matmul_llo_jf.py[Added - diff]
mosaic/examples/matmul_mhlo.py[Added - diff]
mosaic/examples/rope.py[Added - diff]
mosaic/examples/segmented_matmul.py[Added - diff]
mosaic/examples/trmm.py[Added - diff]
mosaic/examples/trmm_llo_jf.py[Added - diff]
mosaic/mlir-tpu-opt.cc[Added - diff]
mosaic/partir_callback_registration.cc[Added - diff]
mosaic/python/BUILD[Added - diff]
mosaic/python/apply_vector_layout_test.py[Added - diff]
mosaic/python/custom_call_emitter.cc[Added - diff]
mosaic/python/dialects/BUILD[Added - diff]
mosaic/python/dialects/_llo_ops_ext.py[Added - diff]
mosaic/python/dialects/llo.py[Added - diff]
mosaic/python/dialects/llo_ext.cc[Added - diff]
mosaic/python/dialects/llo_python.td[Added - diff]
mosaic/python/megacore_adjuster_base.h[Added - diff]
mosaic/python/multidevice_test.py[Added - diff]
mosaic/python/test_util.py[Added - diff]
mosaic/python/tpu_custom_call_test.py[Added - diff]
mosaic/tests/BUILD[Added - diff]
mosaic/tests/build_defs.bzl[Added - diff]
mosaic/tests/lit/cse_opt.mlir[Added - diff]
mosaic/tests/lit/full_unrolling.mlir[Added - diff]
mosaic/tests/lit/lit_test.sh[Added - diff]
mosaic/tests/lit/lower_to_llo.mlir[Added - diff]
mosaic/tests/lit/mhlo_vectorization.mlir[Added - diff]
mosaic/tests/lit/roundtrip.mlir[Added - diff]
mosaic/tests/llo_dump_diff.py[Added - diff]
mosaic/tests/mlir_interpreter.py[Added - diff]
mosaic/tests/sparsecore_persistent_args.py[Added - diff]
mosaic/tests/sparsecore_tile_index.py[Added - diff]
mosaic/tests/sparsecore_vload_arith.py[Added - diff]

58 files changed

tree: 8d0c8edbbb935c2a7c2cf27b216b85eba9ae6b9e