pub struct InverseSqrt {
    pub iterations: u64,
    pub denominator_cap_2k: u64,
}
Expand description

A structure that defines the custom operation InverseSqrt that computes an approximate inverse square root using Newton iterations.

In particular, this operation computes an approximation of 2denominator_cap_2k / sqrt(input).

Input must be of the scalar type UINT64/INT64 and be in (0, 22 * denominator_cap_2k - 1) range. The input is also assumed to be small enough (less than 221), otherwise integer overflows are possible, yielding incorrect results. In case of INT64 type, negative inputs yield undefined behavior.

Optionally, an initial approximation for the Newton iterations can be provided. In this case, the operation might be faster and of lower depth, however, it must be guaranteed that 22 * denominator_cap_2k - 2 <= input * initial_approximation <= 22 * denominator_cap_2k.

The following formula for the Newton iterations is used: x_{i + 1} = x_i * (3 / 2 - d / 2 * x_i * x_i).

Custom operation arguments

  • Node containing an unsigned 64-bit array or scalar to compute the inverse square root
  • (optional) Node containing an array or scalar that serves as an initial approximation of the Newton iterations

Custom operation returns

New InverseSqrt node

Example

let c = create_context().unwrap();
let g = c.create_graph().unwrap();
let t = array_type(vec![2, 3], UINT64);
let n1 = g.input(t.clone()).unwrap();
let guess_n = g.input(t.clone()).unwrap();
let n2 = g.custom_op(CustomOperation::new(InverseSqrt {iterations: 10, denominator_cap_2k: 4}), vec![n1, guess_n]).unwrap();

Fields§

§iterations: u64

Number of iterations of the Newton-Raphson algorithm

§denominator_cap_2k: u64

Number of output bits that are approximated

Trait Implementations§

source§

impl CustomOperationBody for InverseSqrt

source§

fn instantiate( &self, context: Context, arguments_types: Vec<Type> ) -> Result<Graph>

Defines the logic of a custom operation. Read more
source§

fn get_name(&self) -> String

Specifies and returns the name of this custom operation. Read more
source§

impl Debug for InverseSqrt

source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
source§

impl<'de> Deserialize<'de> for InverseSqrt

source§

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer. Read more
source§

impl Hash for InverseSqrt

source§

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher. Read more
1.3.0 · source§

fn hash_slice<H>(data: &[Self], state: &mut H)where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher. Read more
source§

impl PartialEq<InverseSqrt> for InverseSqrt

source§

fn eq(&self, other: &InverseSqrt) -> bool

This method tests for self and other values to be equal, and is used by ==.
1.0.0 · source§

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.
source§

impl Serialize for InverseSqrt

source§

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>where __S: Serializer,

Serialize this value into the given Serde serializer. Read more
source§

impl Eq for InverseSqrt

source§

impl StructuralEq for InverseSqrt

source§

impl StructuralPartialEq for InverseSqrt

Auto Trait Implementations§

Blanket Implementations§

source§

impl<T> Any for Twhere T: 'static + ?Sized,

source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
source§

impl<T> Borrow<T> for Twhere T: ?Sized,

source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
source§

impl<T> BorrowMut<T> for Twhere T: ?Sized,

source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
source§

impl<Q, K> Equivalent<K> for Qwhere Q: Eq + ?Sized, K: Borrow<Q> + ?Sized,

source§

fn equivalent(&self, key: &K) -> bool

Compare self to key and return true if they are equal.
source§

impl<T> From<T> for T

source§

fn from(t: T) -> T

Returns the argument unchanged.

source§

impl<T, U> Into<U> for Twhere U: From<T>,

source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

source§

impl<T> Same<T> for T

§

type Output = T

Should always be Self
source§

impl<T> Serialize for Twhere T: Serialize + ?Sized,

source§

fn erased_serialize(&self, serializer: &mut dyn Serializer) -> Result<Ok, Error>

source§

impl<T, U> TryFrom<U> for Twhere U: Into<T>,

§

type Error = Infallible

The type returned in the event of a conversion error.
source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
source§

impl<T, U> TryInto<U> for Twhere U: TryFrom<T>,

§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
§

impl<V, T> VZip<V> for Twhere V: MultiLane<T>,

§

fn vzip(self) -> V

source§

impl<T> DeserializeOwned for Twhere T: for<'de> Deserialize<'de>,