squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

set_slice3x3 -> set_slice_3x3 (#1784)

Browse Source
This commit is contained in:
Gabriel Wu 2024-09-06 11:24:10 +08:00 committed by GitHub
parent 06e337758d
commit 82f5075946
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
include/cutlass/matrix.h
View File

@ -7825,7 +7825,7 @@ struct Matrix<Element_, 3, 3> {
Matrix m;
m.set_slice3x3({
m.set_slice_3x3({
c + x * x * one_minus_cos, x * y * one_minus_cos - z * s, x * z * one_minus_cos + y * s,
y * x * one_minus_cos * z * s, c + y * y * one_minus_cos, y * z * one_minus_cos - x * s,
z * x * one_minus_cos - y * s, z * y * one_minus_cos + x * s, c + z * z * one_minus_cos
@ -7845,7 +7845,7 @@ struct Matrix<Element_, 3, 3> {
Matrix m = Matrix::identity();
m.set_slice3x3({
m.set_slice_3x3({
Element(1) - Element(2) * a * a, Element(-2) * a * b, Element(-2) * a * c,
Element(-2) * a * b, Element(1) - Element(2) * b * b, Element(-2) * b * c,
Element(-2) * a * c, Element(-2) * b * c, Element(1) - Element(2) * c * c
@ -14005,7 +14005,7 @@ struct Matrix<Element_, 4, 4> {
Matrix m;
m.set_slice3x3({
m.set_slice_3x3({
c + x * x * one_minus_cos, x * y * one_minus_cos - z * s, x * z * one_minus_cos + y * s,
y * x * one_minus_cos * z * s, c + y * y * one_minus_cos, y * z * one_minus_cos - x * s,
z * x * one_minus_cos - y * s, z * y * one_minus_cos + x * s, c + z * z * one_minus_cos
@ -14025,7 +14025,7 @@ struct Matrix<Element_, 4, 4> {
Matrix m = Matrix::identity();
m.set_slice3x3({
m.set_slice_3x3({
Element(1) - Element(2) * a * a, Element(-2) * a * b, Element(-2) * a * c,
Element(-2) * a * b, Element(1) - Element(2) * b * b, Element(-2) * b * c,
Element(-2) * a * c, Element(-2) * b * c, Element(1) - Element(2) * c * c