PR #46339 Extended & simplified n-to-1 kernel fusion via KernelConfig

feat: module fusion API for kernels

michaelbenayoun committed 77 days ago

fix: improve __repr__ for fused modules

michaelbenayoun committed 77 days ago

wip: integration to KernelConfig

michaelbenayoun committed 77 days ago

wip: add temporary example

michaelbenayoun committed 77 days ago

wip: pattern matching in KernelConfig and actual kernel repo

michaelbenayoun committed 74 days ago

refactor: move relevant code to hub_kernels.py

michaelbenayoun committed 74 days ago

docs: reformat docstring

michaelbenayoun committed 74 days ago

refactor: remove comment

michaelbenayoun committed 74 days ago

Merge branch 'main' into fused_kernels

michaelbenayoun committed 74 days ago

Merge branch 'main' into fused_kernels

michaelbenayoun committed 60 days ago

refactor: update example script for testing

michaelbenayoun committed 60 days ago

wip: remove apply_fusions method

michaelbenayoun committed 60 days ago

wip: add core feature for integration with the current fusing API

michaelbenayoun committed 60 days ago

fix: move kernel mapping patching to kernelize

michaelbenayoun committed 59 days ago

wip: update example script

michaelbenayoun committed 59 days ago

wip: add transform_model method for WeightTransform

michaelbenayoun committed 56 days ago

wip: conversion_mapping in Kernel

michaelbenayoun committed 56 days ago

Merge branch 'main' into extended_kernels_api

michaelbenayoun committed 56 days ago

wip: remove things from __all__

michaelbenayoun committed 53 days ago

wip: remove imports

michaelbenayoun committed 53 days ago

fix: remove register_fusion_pattern path

michaelbenayoun committed 53 days ago

Merge branch 'main' into extended_kernels_api

michaelbenayoun committed 53 days ago

fix: remove unused attribute

michaelbenayoun committed 53 days ago

wip: update experimentation script

michaelbenayoun committed 52 days ago

refactor: add convert as abstract method

michaelbenayoun committed 52 days ago

style: reformat hub_kernels.py

michaelbenayoun committed 52 days ago

wip: transform_model API

michaelbenayoun committed 52 days ago

wip: transform_model API, WeightTransform

michaelbenayoun committed 52 days ago

wip: transform_model API, WeightConverter

michaelbenayoun committed 52 days ago

wip: transform_model API, WeightConverter

michaelbenayoun committed 52 days ago

wip: make transform_model idempotent

michaelbenayoun committed 51 days ago

refactor: infer_kernel_fusion_transforms

michaelbenayoun committed 51 days ago

style: regexs -> regexes

michaelbenayoun committed 51 days ago

refactor: register_kernel_fusions

michaelbenayoun committed 51 days ago

refactor: post transformation cleanup

michaelbenayoun committed 51 days ago

Merge branch 'main' into extended_kernels_api

michaelbenayoun committed 51 days ago

style: fix comment

michaelbenayoun committed 50 days ago

test: add TestApplyTransformsToMetaModel tests

michaelbenayoun committed 50 days ago

test: add kernels test

michaelbenayoun committed 50 days ago

Merge branch 'main' into extended_kernels_api

michaelbenayoun committed 50 days ago

test: fix hub_kernels package reload

michaelbenayoun committed 45 days ago

style: ruff

michaelbenayoun committed 45 days ago

Merge branch 'main' into extended_kernels_api

michaelbenayoun committed 45 days ago

refactor: do not create dynamic classes in test

michaelbenayoun committed 45 days ago

refactor: no dynamic class creation in tests

michaelbenayoun committed 45 days ago

refactor: test

michaelbenayoun committed 44 days ago

Merge branch 'main' into extended_kernels_api

michaelbenayoun committed 44 days ago

fix: TYPE_CHECKING imports were broken

michaelbenayoun committed 44 days ago

Merge branch 'main' into extended_kernels_api

michaelbenayoun committed 44 days ago

wip: get rid of transform_model methods

michaelbenayoun committed 38 days ago

wip: move tests

michaelbenayoun committed 38 days ago

wip: make conversion happen before fused module instantiation

michaelbenayoun committed 38 days ago

refactor

michaelbenayoun committed 38 days ago

wip: move conversion_mapping inside the init

michaelbenayoun committed 38 days ago

wip: without any transform_model

michaelbenayoun committed 31 days ago

wip: remove dead code

michaelbenayoun committed 31 days ago

wip: api imrpovement

michaelbenayoun committed 31 days ago

wip: refactor

michaelbenayoun committed 31 days ago

wip: enable __init__ support in kernels