Yes you can link the sale to the K-1 entity.
In the Dispositions Screen 17 go down to Corresponding Activity Information. It works for nonpassive activities, too.
