I've heard that the A998 came behind 318's and the A999's came behind 360's. Not sure if there's any truth to it though. Is the only difference the number of frictions in the direct drum?
Do you have a pic of the input shaft of the transmission? The lockup versions (998-999) input shafts are very noticeably smaller at the end than a plain ol 904. The first half inch or so is smaller than the rest of the shaft.