Upload torch-ext/torch_binding.h with huggingface_hub

e443917 verified 7 months ago

164 Bytes

#pragma once

#include <torch/torch.h>

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

	#pragma once

	#include <torch/torch.h>

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