medmekk
/

rmsnorm_kernel

Model card Files Files and versions

rmsnorm_kernel / torch-ext /torch_binding.h

medmekk's picture

medmekk HF Staff

First builds

6b60385 3 days ago

history blame contribute delete

115 Bytes

	#pragma once

	#include <torch/torch.h>

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