search

microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Refactor handwritten vrelu3 to pull out reusable code #898

Closed dcrc2 closed 3 years ago

dcrc2 commented 3 years ago

The handwritten vrelu3 contained code that can be reused for any elementwise operation. This PR pulls this out into a new header inside src/runtime.

Additionally, the wrapper now asserts that the tensor element type is double, as this is the only floating-point type that Cgen supports.

dcrc2 commented 3 years ago

Additionally, the wrapper now asserts that the tensor element type is double, as this is the only floating-point type that Cgen supports.

This will need updating due to #928

dcrc2 commented 3 years ago

Additionally, the wrapper now asserts that the tensor element type is double, as this is the only floating-point type that Cgen supports.

This will need updating due to #928

Done.