Have not tested with other sledgehammers.
Currently using level 2 iron sledghammer with the burning shaft mod. The flame effect shows, but it does not act as a torch and give off any light.
Burning shaft mod appears to function normal on other tools I've tried.