medmekk
/

rmsnorm_thrust

Model card Files Files and versions Community

rmsnorm_thrust / torch-ext /torch_binding.h

medmekk's picture

medmekk HF Staff

Upload custom kernels

4303459 verified 9 days ago

history blame contribute delete

117 Bytes

	#pragma once

	#include <torch/torch.h>

	void rmsnorm_forward(torch::Tensor const &input, torch::Tensor const &gamma);