triton

gaspersic/triton

Fork 0

Commit Graph

Author	SHA1	Message	Date
daadaada	d8d6b715c8	[CODEGEN] Performance improvement on A100 (#125 ) Improved codegen for the Ampere GPUs. * Make the layout pass recognize the multistage pipelined pattern. * Now the pipeline pass can automate the multistage pipelining transformation. * Remove extra barriers (from the prefetch pass & WAR) on Ampere. * Update the code generator (generator.cc) to make Triton generate n-buffered shared memory loads/stores.	2021-07-27 12:38:49 -07:00
Philippe Tillet	38ab4e955a	[CODEGEN] Bugfix in prefetch pass (#118 )	2021-07-27 12:38:49 -07:00
daadaada	967e629c0c	[CODEGEN] Add a pass to prefetch operands of dot if applicable. (#105 ) * update membar pass when data is double buffered * Add instruction prefetch_s * prefetch tests pass (except the 1 warp case) * Fix the 1-warp bug * Add back prefetch files * Disable prefetch on a100 * Always add war barrier on sm>=80	2021-07-27 12:38:49 -07:00

Author

SHA1

Message

Date

daadaada

d8d6b715c8

[CODEGEN] Performance improvement on A100 (#125 )

Improved codegen for the Ampere GPUs.

    * Make the layout pass recognize the multistage pipelined pattern.
    * Now the pipeline pass can automate the multistage pipelining transformation.
    * Remove extra barriers (from the prefetch pass & WAR) on Ampere.
    * Update the code generator (generator.cc) to make Triton generate n-buffered shared memory loads/stores.

2021-07-27 12:38:49 -07:00

Philippe Tillet

38ab4e955a

[CODEGEN] Bugfix in prefetch pass (#118 )

2021-07-27 12:38:49 -07:00

daadaada

967e629c0c

[CODEGEN] Add a pass to prefetch operands of dot if applicable. (#105 )

* update membar pass when data is double buffered

* Add instruction prefetch_s

* prefetch tests pass (except the 1 warp case)

* Fix the 1-warp bug

* Add back prefetch files

* Disable prefetch on a100

* Always add war barrier on sm>=80

2021-07-27 12:38:49 -07:00

3 Commits