Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
lietorch
Commits
563e40d8
Commit
563e40d8
authored
May 14, 2025
by
fengzch-das
Browse files
Merge branch 'fzc-dev' into 'master'
fzc-dev See merge request
!1
parents
e867de79
1a8ddcca
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
3 deletions
+5
-3
README.md
README.md
+4
-2
lietorch/src/lietorch_gpu.cu
lietorch/src/lietorch_gpu.cu
+1
-1
No files found.
README.md
View file @
563e40d8
...
...
@@ -69,5 +69,7 @@ Type "help", "copyright", "credits" or "license" for more information.
sh run_tests.sh
```
## Known Issue
-
无
## 注意
针对 eigen 需要做下面几处修改:
+
lietorch/eigen/Eigen/src/Core/arch/Default/Half.h:669 行将
``return half(::hlog(a)); 修改为 return half(::hlog(float(a)));``
+
lietorch/eigen/Eigen/src/Core/arch/Default/Half.h:776 行将
``return x + (y-x) * half(float(std::rand()) / float(RAND_MAX));``
修改为
``return half(x + (y-x) * half(float(std::rand()) / float(RAND_MAX)));``
\ No newline at end of file
lietorch/src/lietorch_gpu.cu
View file @
563e40d8
...
...
@@ -215,7 +215,7 @@ __global__ void act_backward_kernel(const scalar_t* grad, const scalar_t* X_ptr,
Point
p
(
p_ptr
+
i
*
3
);
PointGrad
dq
(
grad
+
i
*
3
);
Eigen
::
Map
<
PointGrad
>
(
dp
+
i
*
3
)
=
dq
*
X
.
Matrix4x4
().
block
<
3
,
3
>
(
0
,
0
);
Eigen
::
Map
<
PointGrad
>
(
dp
+
i
*
3
)
=
dq
*
X
.
Matrix4x4
().
template
block
<
3
,
3
>(
0
,
0
);
Eigen
::
Map
<
Grad
>
(
dX
+
i
*
Group
::
N
)
=
dq
*
Group
::
act_jacobian
(
X
*
p
);
}
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment