Grigory Sizov
Grigory Sizov
@googlebot I signed it!
@sanbuphy Yes, bfloat16 is now supported: see https://github.com/facebookincubator/AITemplate/blame/2f61d70c4b14d59f557c1fd9a98cbb5030307f9d/python/aitemplate/compiler/dtype.py#L28
Yes, in theory that should work. I imagine some ops not having implementations for some of the data types though. Let us know if you notice any hiccups
@chsungen That's a valid point - it would be useful to show shapes information when the model is printed out. For now the best way I can think of to...
@sanbuphy Unfortunately AITemplate doesn't support mixed precision/auto-cast functionality. @chenyang78 CC'ing you for visibility, given the recent discussion on the topic
@erjiang Thanks for the PR! A few thoughts: - Tests are breaking at the moment, please have a look - To test that the transform actually works, you might want...
> @ipiszy Can you give me access to [fburl.com/pimcs20r](https://fburl.com/pimcs20r) ? @tchaton This link resolves to https://github.com/facebookincubator/AITemplate#installation , if that helps
We don't have active plans for adding it at the moment, but contributions are welcome. See also https://github.com/facebookincubator/AITemplate/issues/888
hi @scxiao , I'm on vacation and travelling till May 11 - I can review it after that, or @bottler will have a look in the meantime On Mon 6....
> QQ: do we plan to merge this PR as it has been pending for months. Sorry, I didn't follow-up on @tridao's comment above. Basically I think there should be...