11 Commits

Author	SHA1	Message	Date
Keren Zhou	328b87aec6	Keren/tensor slice insert alloc (#94) ... This branch defines three new triton_gpu operations to partially solve #87. Below is an overview: ``` %tensor = triton_gpu.alloc_tensor : tensor<2x16x16xf16, #A> %b = triton_gpu.insert_slice_async %a_ptr, %tensor, %offset {axis = 0 : i32, cache = 1 : i32, evict = 1 : i32, isVolatile = false} : tensor<16x16x!tt.ptr<f16>, #AL> -> tensor<2x16x16xf16, #A> %c = triton_gpu.extract_slice %b, %offset {axis = 0 : i32} : tensor<2x16x16xf16, #A> -> tensor<16x16xf16, #A> ``` We plan to fully replace `copy_async` with `insert_slice_async`. **This hasn't been done yet.**	2022-09-01 12:37:17 -07:00
Keren Zhou	02ebf24d35	Analyze shared memory alias (#81) ... The purpose of this PR is analyzing shared memory aliases so that we can fix memory allocation bugs and save memory allocations in triton code involving complex control flows. Changes to memory bar and allocation are on the way. Co-authored-by: Philippe Tillet <phil@openai.com>	2022-08-29 10:43:20 -07:00
Keren Zhou	e0bedeb44c	[BACKEND] Keren/shared memory barrier (#59)	2022-08-18 12:32:57 -07:00
Keren Zhou	a7b49b3227	[BACKEND] Memory allocation (#33)	2022-08-04 11:22:49 -07:00
Yan Chunwei	b988bae813	Init TritonGPU to LLVM dialect conversion (#32) ... * add toLLVM pass * update num-warps setting in mlir	2022-08-04 10:15:45 +08:00
Philippe Tillet	6d62d88d4f	[CI] run clang-format (#24)	2022-07-26 17:25:03 -07:00
Philippe Tillet	a633d2b403	[Analysis] Added Axis Info Analysis (#8)	2022-07-19 13:38:48 -07:00
Yan Da	560e29229b	register conversion in triton-opt	2022-06-07 19:33:51 +08:00
Yan Da	366dddc3bc	update mma encoding & triton-opt	2022-06-06 21:03:58 +08:00
Yan Da	d5eca56cf3	more TritonGPU unit tests	2022-06-05 14:25:09 +08:00
Yan Da	55cf9a0a97	Add triton's opt	2022-06-04 22:10:00 +08:00