It should just work similarly to how Warframe's Mod system works. You can use the mods on the right pieces regardless of element, but matching the element would reduce the cost. Matching the element would provide a bonus, but not be mandatory.