burtenshaw
/

ptc-optimized-kernel

Model card Files Files and versions

ptc-optimized-kernel / torch-ext /torch_binding.h

burtenshaw's picture

burtenshaw HF Staff

Upload torch-ext/torch_binding.h with huggingface_hub

e443917 verified about 2 months ago

history blame contribute delete

164 Bytes

	#pragma once

	#include <torch/torch.h>

	void rmsnorm(
	torch::Tensor& output,
	const torch::Tensor& input,
	const torch::Tensor& weight,
	float eps
	);