Semiconductor equipment maker KLA is seen benefiting from chipmakers buying advanced gear for producing AI chips.