drazi fengxie

fengxie / pass_list_as_vector.py

Created September 10, 2025 09:04

CuTe DSL pass list from python and convert to vector for kernel


	from typing import List

	import cutlass
	import cutlass.cute as cute
	from cutlass.cute.runtime import from_dlpack


	@cute.kernel
	def kernel_use_vec_as_arg(vec, res: cute.Tensor):

fengxie / vector_as_kernel_arg.py

Last active September 14, 2025 17:12

CuTe DSL passing vector as kernel argument


	import cutlass
	import cutlass.cute as cute
	from cutlass.cute.runtime import from_dlpack


	@cute.kernel
	def kernel_use_vec_as_arg(vec, res: cute.Tensor):
	# cute.print_tensor(vec)
	res.store(vec)

fengxie / vscode-config

Created July 24, 2019 09:41

my gist

fdf