Skip to content
Discussion options

You must be logged in to vote

we were actually thinking of removing filter_fn in the future and have people express everything through config (FqnToConfig) itself, and users can generate the config with code:

fqn_to_config = {}
for n, m in model.named_modules():
      if isinstance(m, torch.nn.Conv2d):
          fqn_to_config[n] = conv_config

torchao_config = FqnToConfig(fqn_to_config)

any concerns for this?

also we have limited support for conv for now I think, only on B200 and above

Replies: 1 comment 3 replies

Comment options

You must be logged in to vote
3 replies
@Freed-Wu
Comment options

@jcaip
Comment options

jcaip Jan 13, 2026
Collaborator

@Freed-Wu
Comment options

Answer selected by Freed-Wu
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
3 participants